有没有办法在决策树分类器中同时获得每个类别的预测概率以及每个类别的标签?

时间:2018-07-09 21:01:00

标签: python

我想获取每个类别的预测概率以及决策树分类器的类别标签。

probs = clf.predict_proba(test)

这给了我几率,我想排序以获得前三名,因此下面的代码将给我:

best_n = np.argsort(probs, axis=1)[-n:]

但是我不知道如何将clf.classes_与之结合在一起以将它们整合在一起?

1 个答案:

答案 0 :(得分:0)

看起来像这样既可以对概率进行排序,又可以获取类标签

parts = pd.DataFrame(np.argsort(clf.predict_proba(X_test), axis=1)[-3:], columns=clf.classes_)