怎么用joblib保存模型？

詹惠儿

2018-12-19 阅读量: 1873

怎么用joblib保存模型？

当我们在某个不同的项目中或稍后需要相同的训练数据时，为了避免浪费训练时间，存储经过训练的模型，以便将来可以随时使用。

我们可以通过以下方式保存模型：

使用joblib作为文件的腌制模型：Joblib是pickle的替代品，因为它对携带大型numpy数组的对象更有效。这些函数也接受类文件对象而不是文件名。

joblib.dump序列化对象层次结构
joblib.load以反序列化数据流

使用joblib保存到pickle文件 -

from sklearn.externals import joblib

# Save the model as a pickle in a file

joblib.dump(knn, 'filename.pkl')

# Load the model from the file

knn_from_joblib = joblib.load('filename.pkl')

# Use the loaded model to make predictions

knn_from_joblib.predict(X_test)

0.0000

关注作者

发表评论

暂无数据

CDA考试动态

CDA报考指南

推荐帖子