模型训练完毕后,会因各种需求要将模型做一个保存,例如:在训练集中进行交叉验证或者多模型比较需要将模型保存下来后做一个对比,那么这时就需要用到externals模块保存训练模型。
保存模型:
from sklearn.externals import joblib
## 通过joblib的dump可以将模型保存到本地
joblib.dump(model_name, "savepath.m")
load模型:
#模型从本地调回
model = joblib.load("savepath.m")
#在测试集上测试
model.predit(testData) # testing_X为测试集