很多模型训练完成之后,可以进行保存,下次使用时直接调用即可,不需要再次训练数据。接下来我将介绍sklearn中模型的保存和加载。
sklearn保存模型的api
from sklean
.externals
import joblib
joblib
.dump
(lr
,"./tmp/test.pkl")
lr2
=joblib
.load
("./tmp/test.pkl")
保存数据和导入数据都是训练好的模型,里面包含了模型的参数。
例如:
lr
= LinearRegression
()
lr
.fit
(x_train
, y_train
)
joblib
.dump
(lr
,"./tmp/test.pkl")
lr2
=joblib
.load
("./tmp/test.pkl")
lr2
.fit
(x_train
, y_train
)
转载请注明原文地址:https://blackberry.8miu.com/read-27924.html