让 Scikit-Learn RandomForestClassifier 输出前 N 个结果答案

【问题标题】：Getting Scikit-Learn RandomForestClassifier to output Top N results让 Scikit-Learn RandomForestClassifier 输出前 N 个结果
【发布时间】：2015-04-18 12:49:41
【问题描述】：

我想查看 RandomForestClassifier 预测的前 N 个结果，按概率降序排列。

答案可能是 predict_proba，但我不知道如何解释结果。

帮助表示赞赏！

【问题讨论】：

前 N 个什么？ RFC 是一个二元分类器，它预测观察属于正类的概率。如果你想找到最可能的样本，它完全独立于 sklearn，你只需对概率进行排序。

标签： machine-learning scikit-learn classification prediction random-forest

【解决方案1】：

我认为您使用predict_proba 是对的。使用np.argsort 解释结果：

p = rfc.predict_proba(X)
n = 3
top_n = np.argsort(p)[:,:-n-1:-1]

【讨论】：

如果你的类是从 0 到 n_classes 的整数；）否则你需要使用rfc.classes_[top_n]