有没有办法在决策树分类器中同时获得每个类别的预测概率以及每个类别的标签？
有没有办法在决策树分类器中同时获得每个类别的预测概率以及每个类别的标签？
提问于 2018-07-10 05:01:00
我想要获得每个类别的预测概率，以及决策树分类器的类别标签。
probs = clf.predict_proba(test)。
这给了我概率，我想排序得到前3名，所以下面的代码将给我提供：
best_n = np.argsort(probs, axis=1)[-n:]
但我不知道如何将clf.classes_与此结合在一起才能将它们整合在一起？
回答 1
Stack Overflow用户
回答已采纳
发布于 2018-07-10 06:35:47
看起来这样做既可以对概率进行排序，也可以获得类标签
parts = pd.DataFrame(np.argsort(clf.predict_proba(X_test), axis=1)[-3:], columns=clf.classes_)
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/51253953
复制相关文章
相似问题