训练模型的保存与加载

发布时间:2025-05-24 16:47:02 作者:益华网络 来源:undefined 浏览量(1) 点赞(1)
摘要:1.目的: 将训练好的模型保存下来,已备下次使用,节省训练时间,提高效率 2.API: from sklearn.externals import joblib 保存: joblib.dump(rf,"test.pkl")

  1.目的:

  将训练好的模型保存下来,已备下次使用,节省训练时间,提高效率

  2.API:

  from sklearn.externals import joblib

  保存:

  joblib.dump(rf,"test.pkl")

  加载:

  estimator = joblib.load("test.pkl")

  3.Python代码实现:

  # -*- coding: UTF-8 -*-

  @Author :Jason

  波士顿房价预测,将模型保存到

  from sklearn.datasets import load_boston

  from sklearn.model_selection import train_test_split

  from sklearn.preprocessing import StandardScaler

  from sklearn.linear_model import Ridge

  from sklearn.metrics import mean_squared_error

  from sklearn.externals import joblib

  def model_save_fetch():

  """

  岭回归对波士顿房价进行预测

  :return:

  """

  # 1)获取数据

  boston = load_boston()

  print("特征数量:\n", boston.data.shape)

  # 2)划分数据集 郑州妇科医院哪家好 http://fk.zyfuke.com/

  x_train, x_test, y_train, y_test = train_test_split(boston.data, boston.target, random_state=22)

  # 3)标准化

  transfer = StandardScaler()

  x_train = transfer.fit_transform(x_train)

  x_test = transfer.transform(x_test)

  # # 4)预估器

  # estimator = Ridge(alpha=0.5, max_iter=10000)

  # estimator.fit(x_train, y_train)

  #

  # # 保存模型

  # joblib.dump(estimator, "./files/test.pkl")

  # 加载模型

  estimator = joblib.load("./files/test.pkl")

  # 5)得出模型

  print("岭回归-权重系数为:\n", estimator.coef_)

  print("岭回归-偏置为:\n", estimator.intercept_)

  # 6)模型评估

  y_predict = estimator.predict(x_test)

  print("预测房价:\n", y_predict)

  error = mean_squared_error(y_test, y_predict)

  print("岭回归-均方误差为:\n", error)

  return None

  if __name__ == "__main__":

  model_save_fetch()

二维码

扫一扫,关注我们

声明:本文由【益华网络】编辑上传发布,转载此文章须经作者同意,并请附上出处【益华网络】及本页链接。如内容、图片有任何版权问题,请联系我们进行处理。

感兴趣吗?

欢迎联系我们,我们愿意为您解答任何有关网站疑难问题!

您身边的【网站建设专家】

搜索千万次不如咨询1次

主营项目:网站建设,手机网站,响应式网站,SEO优化,小程序开发,公众号系统,软件开发等

立即咨询 15368564009
在线客服
嘿,我来帮您!