首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >随机森林提高准确率

随机森林提高准确率
EN

Stack Overflow用户
提问于 2020-12-18 06:33:37
回答 1查看 54关注 0票数 2

我正在使用RandomForestClassifier方法进行物体检测,问题是,即使我知道我的随机状态应该是0作为默认值,我得到了非常糟糕的准确性,所以无论如何,有没有知道我的n_estimators,random_state参数的最佳值是什么?

代码语言:javascript
复制
from sklearn.ensemble import RandomForestClassifier
RF_model = RandomForestClassifier(n_estimators = 250, random_state = 120)
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2020-12-18 12:22:14

要确定模型的最佳参数,可以使用称为网格搜索的过程。Sklearn提供了一个用于执行此操作的类GridSearchCV。我已经提供了如何将其用于随机森林分类器的代码示例。

代码语言:javascript
复制
from sklearn.ensemble import RandomForestClassifier
from sklearn.model_selection import GridSearchCV

# provide iterables of values to be tested each parameter
parameters = {'n_estimators': [100, 250, 500, 750]}
clf = GridSearchCV(RandomForestClassifier(), parameters)
clf.fit(X, y)  # X and y are your training data and targets

值得注意的是,在您的问题中,您特别提到要为n_estimatorsrandom_state参数寻找最佳值。我没有将random_state作为GridSearch的一部分,因为该参数通常是为了结果的重现性而存在。下面是Sklearns Glossary中关于该参数的一些additonal reading

票数 2
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/65349290

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档