预测多类分类中的第二类

Question

您好，我已经准备好一个多类分类模型并在数据集上进行了训练

Label    Feat1  Feat2 Feat3 Feat4 
Class1    10     21    12     2
Class2    3      6     7      9
Class3    14     8     8     10
Class4    1      5     5      9

我目前可以使用 Sckit-Learn 中的 predict 函数来应用最佳模型来预测单个类。所以我可以得到列 Predicted_Label。如何解决问题以获得预测列表，即第二或第三最佳预测

Test_Data_Set 
Feat1  Feat2   Feat3 Feat4    Predicted_Label   Predicted_Label_2nd_Best_Prediction
 1       3     10     7          Class1           [Class1,Class4]

Answer 1

请参考这个问题：Understanding predict_proba from MultiOutputClassifier

您需要在您的model 上使用predict_proba() 来获取每个班级、每一行训练数据集的概率。在您的情况下，如果您有 4 个类，您将获得一个 长度为 4 的数组。

然后，您可以获得第二大概率的第二好的预测类。

Multiclass MultiOutput Classification example on sklearn documentation

注意：predict_proba() 中每个长度为 4 的数组加起来为 1。