sklearn训练好的模型,有什么方法可以以某种文件的形式保存下来?以后用的时候就不需要在train fit了,直接读取模型,然后predict。
2个回答
#保存
from sklearn.externals import joblib
joblib.dump(model, 'path/to/my_model.pkl')
#读取
model = joblib.load('path/to/my_model.pkl')
也可以直接用pickle将模型保存到本地
import pickle
with open('filename.pkl', 'wb') as f:
pickle.dump(clf, f)
读取
with open('filename.pkl', 'rb') as f:
clf = pickle.load(f)