我想获取每个类别的预测概率以及决策树分类器的类别标签。
probs = clf.predict_proba(test)
。
这给了我几率,我想排序以获得前三名,因此下面的代码将给我:
best_n = np.argsort(probs, axis=1)[-n:]
但是我不知道如何将clf.classes_
与之结合在一起以将它们整合在一起?
答案 0 :(得分:0)
看起来像这样既可以对概率进行排序,又可以获取类标签
parts = pd.DataFrame(np.argsort(clf.predict_proba(X_test), axis=1)[-3:], columns=clf.classes_)