sklearn的模型如何保存下来?

  统计/机器学习 Python    浏览次数: 2423
4

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


 

PG Two   2017-09-27 20:25



   2个回答 
11


#保存

from sklearn.externals import joblib

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

#读取

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


参阅: Model persistence

s3040608090   2017-09-28 09:03

4

也可以直接用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)


sasa   2017-10-12 05:27



  相关主题

查看sklearn版本   1回答

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

sklearn里learning_rate什么意思?   2回答

sklearn SGDClassifier的partial_fit是什么意思?   2回答

sklearn.linear_model.SGDClassifier能做小批量学习(mini-batch)吗?   2回答

sklearn里的LabelEncoder什么用?   3回答

sklearn predict的默认阈值   2回答

sklearn.linear_model.SGDClassifier中的参数n_iter设置问题   2回答

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

sklearn中的predict_proba方法的返回值的意义   2回答

sklearn.svm.SVC中的gamma和rbf中的sigma什么关系   1回答

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



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

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

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

我们谢绝答非所问。

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

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