我想查看 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/