将python随机森林模型保存到⽂件你说的问题叫模型持久化,就是把学习好的模型保存起来,以后只要调⽤这个⽂件就可以了。
每个框架都应该有模型持久化函数,以sklearn为例:
als import joblib
joblib.dump(clf, "train_model.m") #存储
clf = joblib.load("train_model.m") #调⽤
...
import cPickle
rf = RandomForestRegresor()
rf.fit(X, y)
with open('path/to/file', 'wb') as f:
cPickle.dump(rf, f)import pickle
# in your prediction file
with open('path/to/file', 'rb') as f:
rf = cPickle.load(f)
preds = rf.predict(new_X)
版权声明:本站内容均来自互联网,仅供演示用,请勿用于商业和其他非法用途。如果侵犯了您的权益请与我们联系QQ:729038198,我们将在24小时内删除。
发表评论