显示概率和目标值的机器学习预测答案

【问题标题】：Machine Learning Prediction with Probability and Target Value displayed显示概率和目标值的机器学习预测
【发布时间】：2020-10-08 10:40:14
【问题描述】：

我创建了一个多分类 SVC 模型，该模型基于文本将输入文本分类为约 100 个目标值之一。

我已经成功展示了 Predict_proba：

输出：

array([[0.050, 0.001, 0.002, 0.701, 0.032, 0.008, 0.002, 0.001, 0.002,
        0.003, 0.005, 0.001 .... ]])

但是，我还希望输出只显示它所属的目标变量。我还希望它首先输出概率最高的目标变量。它可能最终看起来像这样：

输出：Target4: 0.701, Target1: 0.050, Target5: 0.032 等.....

我怎样才能做到这一点？

只显示前 10 个预测目标值也很棒（所以我没有得到包含约 100 个目标值概率的输出）。

【问题讨论】：

【解决方案1】：

如果您使用的是sklearn，您可以尝试使用此 sn-p 打印顶级 N 类及其相应概率的列表：

print(sorted(zip(clf.classes_, probs[0]), key=lambda x:x[1])[-N:])

【讨论】：