无法在Scikit Learn中将class_weight分配给RandomForestClassifier

时间:2015-08-04 08:35:53

标签: python scikit-learn

我刚才开始使用scikit学习包在我的数据集上实现随机森林。我试图基于多个类创建一个模型,并尝试实现RandomForestClassifier。但是,我认为我有一些不平衡,我想使用class_weight =" auto"参数:

RFC = RandomForestClassifier(n_estimators = int(trees),class_weight="auto").fit(X_train, y_train)

然而,当我尝试运行它时,我得到了

__init__() got an unexpected keyword argument 'class_weight'

我尝试检查其他问题,因为我认为我没有使用正确的表示法,但他们似乎都参考了class_weight =" auto"就这样。

注意:RF不使用class_weight参数。我只是想尝试改进我的结果,因为我认为数据是不平衡的。

谢谢(如果我在格式化或问题上做错了,我会编辑它,第一个问题在这里)

1 个答案:

答案 0 :(得分:1)

我错误地检查了错误的版本列表。我在ipython中运行,虽然我在服务器上更新了它,但它没有在ipython环境中进行,当我用conda检查时,它一直没有ipython环境。 我更新了它,它工作,谢谢。 对不起,但感谢您对此进行调查。