gpt4 book ai didi

python - 从 OneVsRestClassifier 获取随机森林 feature_importances_ 用于多标签分类

转载 作者:太空狗 更新时间:2023-10-30 02:44:02 27 4
gpt4 key购买 nike

我正在使用 OneVsRestClassifier 解决多标签分类问题。我将 RandomForestClassifier 传递给它。

from sklearn.multiclass import OneVsRestClassifier
from sklearn.ensemble import RandomForestClassifier
clf = OneVsRestClassifier(RandomForestClassifier(random_state=0,class_weight='auto',min_samples_split=10,n_estimators=50))
clf.fit(train,dv_train)
print clf.feature_importances_
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
AttributeError: 'OneVsRestClassifier' object has no attribute 'feature_importances_'

如何获取 OneVsRestClassifier 中每个随机森林的特征重要性?

最佳答案

OneVsRestClassifier 有一个属性 estimators_ :n_classes 估计器列表

所以要得到第i个RandomForest

的特征重要性
print clf.estimators_[i].feature_importances_

关于python - 从 OneVsRestClassifier 获取随机森林 feature_importances_ 用于多标签分类,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/31137011/

27 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com