随机森林模型保存-python

来源:互联网 发布:美国 非农 数据 美元 编辑:程序博客网 时间:2024/04/28 10:18

训练完的随机森林模型需要保存下来,目前给出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可能是一个更好的选择


from sklearn.externals import joblib

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

0 0
原创粉丝点击