在keras中是否存在等价函数py手电筒的损失函数nn.crossEntropyLoss()？

Question

我正在将conv2d模型转换为keras(Tensorflow 2.0)。

我快做完了，但最后一个问题是拖延我的工作。

标题说明了这一切:在keras中是否有等效函数py手电筒的丢失函数nn.crossEntropyLoss()？

谢谢你的帮助。

Answer 1

根据标签和输出的类型，Keras中目前支持三种不同的熵损失。

categorical_crossentropy：

keras.losses.categorical_crossentropy(y_true, y_pred, from_logits=False, label_smoothing=0)

sparse_categorical_crossentropy

keras.losses.sparse_categorical_crossentropy(y_true, y_pred, from_logits=False, axis=-1)

binary_crossentropy

keras.losses.binary_crossentropy(y_true, y_pred, from_logits=False, label_smoothing=0)

您可以找到Keras损失的完整列表这里。