2015-06-30 111 views
2

我正在使用OneVsRestClassifier处理多标签分类问题。我将RandomForestClassifier传入它。从OneVsRestClassifier获取随机森林feature_importances_用于多标签分类

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中每个随机森林的功能重要性?

回答

2

OneVsRestClassifier有一个属性estimators_:n_classes的名单估计器

因此得到i个随机森林

print clf.estimators_[i].feature_importances_ 
的功能重要性