随机森林模型保存-python

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/xiangcheng001/article/details/52755739

训练完的随机森林模型需要保存下来，目前给出python的方案。

import pickle

with open("model.pkl", "wb") as f:

pickle.dump(model, f)

# 如果需要再取出来

with open("data.pkl", "rb") as f:

model = pickle.load(f)

据说对于sklearn，joblib可能是一个更好的选择

扫描二维码关注公众号，回复： 5067819 查看本文章

from sklearn.externals import joblib

joblib.dump(clf, 'filename.pkl')