热线电话:13121318867

登录
2020-08-11 阅读量: 1154
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
2
关注作者
收藏
评论(0)

发表评论

暂无数据