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

Posted 2023-03-12

【中文标题】让 Scikit-Learn RandomForestClassifier 输出前 N 个结果

【英文标题】：Getting Scikit-Learn RandomForestClassifier to output Top N results

【发布时间】：2015-04-18 12:49:41

【问题描述】：

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

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

帮助表示赞赏！

【问题讨论】：

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

【参考方案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]