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

Question

我正在將 pytorch 編寫的 conv2d 模型轉換為 keras（Tensorflow 2.0）。

我已經快完成了，但最后一個問題是推遲我的工作。

標題說明了一切：keras 中是否有等效的函數 pytorch 的損失函數 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)

二元交叉熵：

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

您可以在此處找到 Keras 損失的完整列表