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

【问题标题】：Is there an equivalent function pytorch's loss fuction nn.crossEntropyLoss() in keras?keras中是否有等效的函数pytorch的损失函数nn.crossEntropyLoss()？
【发布时间】：2019-11-26 08:45:37
【问题描述】：

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

我差不多完成了，但最后一个问题是耽误我的工作。

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

感谢您的帮助。

【问题讨论】：

您在寻找tf.keras.losses.CategoricalCrossentropy吗？ tensorflow.org/api_docs/python/tf/keras/losses/…

标签： python tensorflow keras pytorch

【解决方案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 损失列表here

【讨论】：

binary_crossentropy 肯定和 pytorch 的 CrossEntropyLoss 不一样。相当于pytorch的BCELoss。