tflearn中的稀疏分类交叉熵损失

时间:2019-03-01 16:44:12

标签: keras tflearn cross-entropy

我正在将https://www.tensorflow.org/tutorials/中的mnist示例转换为从keras那里学习。但是,在tflearn中,没有sparse_categorical_crossentropy,只有categorical_crossentropy。因此,我必须将所有y向量(它们只是一个整数1 ... 10,指定数据集中图像所属的类)为格式为[0 ... 1 ... 0]的1x10向量

在Keras中,我可以只在sparse_categorical_crossentropy和categorical_crossentropy之间切换,这取决于目标是索引的整数还是[0 0 ... 1 ... 0]形式的向量。

tflearn中是否有类似的损失函数?或者categorical_crossentropy是唯一的选择吗?

0 个答案:

没有答案