使用sklearn时搞清楚类的标签MLP.predict_proba（）函数

Question

我正在创建带标记类的神经网络，它们是二进制值列表，并且有1167个类。我想根据预测概率，以利用MLP.predict_proba（）到输出的前5类，但输出的是NP阵列的概率为每个类只由索引值标记，即

enter image description here

我想找出哪些类与这些概率相关联，并且预测函数能够正确输出代表它所预测类的二进制列表。无论如何，我可以手动将这些类标记为这些唯一的二进制列表吗？我基本上有标记类，我的神经网络能够输出正确的二进制列表作为它的预测，就在我使用predict_proba（）时，它只给出了概率，并没有明确指示这些概率对于哪些类。谢谢！

Answer 1

安装后使用mlp的classes_。按照the documentation：

classes_：阵列或形状的阵列的列表（n_classes，）类别标签 每个输出。

它会打印出从数据集中学到的类（大部分按字母顺序）。

predict_proba()将以相同的顺序输出值。