将Python随机森林模型保存到文件

在R语言中,运行完“随机森林”模型后,我可以使用save.image("***.RData")来存储模型。之后,我可以直接加载模型进行预测。

在Python中可以做类似的事情吗?我将模型和预测分成了两个文件。在模型文件中:

rf= RandomForestRegressor(n_estimators=250, max_features=9,compute_importances=True)fit= rf.fit(Predx, Predy)

我尝试返回rffit,但仍然无法在预测文件中加载模型。

你能使用sklearn的随机森林包将模型和预测分开吗?


回答:

...import cPicklerf = RandomForestRegresor()rf.fit(X, y)with open('path/to/file', 'wb') as f:    cPickle.dump(rf, f)# 在你的预测文件中                                                                                                                                                                                                           with open('path/to/file', 'rb') as f:    rf = cPickle.load(f)preds = rf.predict(new_X)

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注