D.Chu

2020-08-11   阅读量: 879

python 保存模型的两种方法

扫码加入数据分析学习群

(1)使用joblib

from sklearn.externals import joblib

保存模型:joblib.dump(reg,'regtree.pkl')

读取模型:clf = joblib('regtree.pkl')


(2)使用pickle

import pickle

保存模型:with open('reg.pkl','wb') as f:

pickle.dump(clf,f)

读取模型:with open('clf.pkl','rb') as f:

new_clf = pickle.load(f)

new_clf

35.2534 4 2 关注作者 收藏

评论(0)


暂无数据

推荐课程