python:如何使用Random Forest了解每个类的重要功能

时间:2015-05-20 23:26:28

标签: python random-forest feature-selection

我使用sklearn.ensemble.RandomForestClassifier进行分类。我总共有14个班级(14个标签)。现在我的代码就像

clf = RandomForestClassifier(n_estimators = 50)
clf.fit(train_data, labels)
importances = clf.feature_importances_
indices = np.argsort(importances)[-20:]
通过这样做,我可以得到前20个重要特征的指数。但是,因为我有14个班级(0,1,2,3,4,5,6,7,8,9,10,11,12,13),现在我想知道每个班级,最重要的是什么功能是,那我该怎么办?

0 个答案:

没有答案