sklearn的模型如何保存下来?

  统计/机器学习 Python    浏览次数:3629        分享
4

sklearn训练好的模型,有什么方法可以以某种文件的形式保存下来?以后用的时候就不需要在train fit了,直接读取模型,然后predict。


 

PG Two   2017-09-27 20:25



   2个回答 
12


#保存

from sklearn.externals import joblib

joblib.dump(model, 'path/to/my_model.pkl')

#读取

model = joblib.load('path/to/my_model.pkl') 


参阅: Model persistence

SofaSofa数据科学社区DS面试题库 DS面经

s3040608090   2017-09-28 09:03

5

也可以直接用pickle将模型保存到本地

import pickle
with open('filename.pkl', 'wb') as f:
    pickle.dump(clf, f)

读取

with open('filename.pkl', 'rb') as f:
    clf = pickle.load(f)


SofaSofa数据科学社区DS面试题库 DS面经

sasa   2017-10-12 05:27



  相关主题

sklearn训练classifier的时候报错Unknown label type   1回答

求sklearn.pipeline.Pipeline使用的例子   1回答

sklearn onehotencoder的具体用法   2回答

sklearn实现adaboost算法了吗?   1回答

sklearn predict的默认阈值   2回答

sklearn里的LabelEncoder什么用?   3回答

当训练集只有一个特征的时候,sklearn的LinearRegression报错   1回答

lightgbm.LGBMModel和lightgbm.LGBMClassifier该用哪个?   1回答

sklearn计算MAPE   1回答

python sklearn模型中random_state参数的意义   2回答

sklearn OneHotEncoder string类型报错   2回答

sklearn里有没有可以求kappa值的?   1回答



回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单,你自己想”、“书上有的,你认真看”这类的回答;如果你认为对方的提问方式或者内容不妥,你可以直接忽略该问题,不用进行任何作答,甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出!