让Scikit Learn RandomForest分类器输出前N个结果



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

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

感谢帮助!

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

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

相关内容

  • 没有找到相关文章

最新更新