从 Keras 顺序模型中恢复原始类标签



我是keras的新手,并且已经训练了一个简单的分类数据顺序模型。我一个热编码标签并在一个热编码上训练模型。但是,当我使用 predict_classes 时,我得到了一个整数数组(我知道它们表示预测返回的 softmax 输出中表示的最可能的类的索引(,但我不知道如何将该输出映射回原始的热编码。我似乎无法在网上或文档中找到答案。任何帮助将不胜感激。

事实证明,keras.predict返回的数组值映射到一个热编码的可能性,其有效数字在该索引处(即prediction[0]表示标签[1, 0, 0, 0]的可能性,prediction[1]表示标签[0, 1, 0, 0]等的可能性(。keras.predict_class返回的值还映射到一个热编码的标签,其有效数字位于该索引处。您可以使用以下方法轻松地从一个热编码移动到 keras 的编码

keras_encoding = np.argmax(one_hot_encoding)

从 keras 的编码到一个热编码

one_hot_encoding = np.zeros(len(single_label))
one_hot_encoding[keras_encoding] = 1

最新更新