我想看看RandomForestClassifier预测的前N个结果,按降序排列。
答案可能是predict_proba,但我不知道如何解释结果。
感谢帮助!
我认为您使用predict_proba
是正确的。使用np.argsort
解释结果:
p = rfc.predict_proba(X)
n = 3
top_n = np.argsort(p)[:,:-n-1:-1]