随机森林模型保存-python

训练完的随机森林模型需要保存下来,目前给出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')

你可能感兴趣的:(机器学习,python,随机森林)