使用sklearn MLP.predict_proba（）函数时确定类的标签

Question

我正在创建一个带有标记类的神经网络，这些类是二进制值列表，共有1167个类。我想利用MLP.predict_proba（）根据预测概率输出前5个类，但输出是一个np数组，每个类的概率只用索引值标记，即

enter image description here

我想弄清楚哪些类与这些概率相关联，并且预测函数正确输出表示预测类的二进制列表。反正我是否可以手动将类标记为这些唯一的二进制列表？我基本上都有标记类，我的神经网络能够输出正确的二进制列表用于预测，就在我使用predict_proba（）时，它只给出了概率，没有明确指出这些概率的类别。谢谢！

Answer 1

安装后使用mlp的classes_。根据{{​​3}}：

  

classes_：数组或形状数组的列表（n_classes，）类标签   对于每个输出。

它将打印出从数据集中学习的类（主要按字母顺序排列）。

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