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

时间:2018-06-13 09:35:00

标签: python keras

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

1 个答案:

答案 0 :(得分:1)

事实证明,keras.predict返回的数组的值映射到一个热编码的可能性,其中该索引具有有效数字(即prediction [0]表示标签[1,0]的可能性, 0,0],预测[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