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

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

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

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

感谢帮助!

最佳答案

我认为您使用predict_proba是正确的。使用np.argsort解释结果:

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

关于machine-learning - 让 Scikit-Learn RandomForestClassifier 输出前 N 个结果,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/28568034/

相关文章:

python - sklearn GaussianNB - 结果不好,[nan] 概率

python - sklearn中train_test_split随机定义训练大小

maven - 根据配置文件更改部署的 Artifact 名称

matlab - 如何在 Matlab 中使用经过训练的神经网络在真实系统中进行分类

java - 如何用java构建贝叶斯信念网络?有图书馆吗? API?

python - 为什么对 `fit` 的第二个 `GridSearchCV` 调用会无休止地工作?

python - 为什么我不能得到与 GridSearchCV 相同的结果?

python - 导入错误: cannot import name TruncatedSVD

classification - 使用这两种方法对 light gbm 分类器有什么区别?

python - 在 ngrams 上训练朴素贝叶斯分类器