sklearn中模型的保存与读取

0. 前言

sklearn 模型在 .fit() 后会得到训练后模型，为了节省后续调用此模型的时间，希望能够将模型保存，下次使用时读取即可。

1. sklearn 模型保存

from sklearn.externals import joblib
#import joblib #也可以直接用这个
joblib.dump(模型变量,'文件名.pkl')

2. sklearn 模型读取

from sklearn.externals import joblib
#import joblib #也可以直接用这个
变量 = joblib.load('文件名.pkl')

不跨平台使用时，可以用joblib.dump进行模型持久化为pkl文件;
跨平台使用时可以根据sklearn2pmml转成pmml文件，通过jpmml去部署到线上。
跨平台可参考：Java PMML API