模型保存和加载

当训练或者计算好一个模型之后,那么如果别人需要我们提供结果预测,就需要保存模型(主要是保存算法的参数)

sklearn模型的保存和加载API

from sklearn.externals import joblib

  • 保存:joblib.dump(rf, 'test.pkl')
  • 加载:estimator = joblib.load('test.pkl')

 

原文地址:https://www.cnblogs.com/a155-/p/14413821.html