python - keras中是否有等效的函数pytorch的损失函数nn.crossEntropyLoss()？

Question

我正在将 pytorch 编写的 conv2d 模型转换为 keras（Tensorflow 2.0）。

我几乎完成了，但最后一个问题是耽误我的工作。

标题说明了一切：在keras中是否有一个等效的函数pytorch的损失函数nn.crossEntropyLoss()？

谢谢你的帮助。

标签： pythontensorflowkeraspytorch

Answer 1

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

分类交叉熵：

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)

二进制交叉熵：

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

你可以在这里找到完整的 Keras 损失列表