keras中是否存在等效函数pytorch的损失函数nn.crossEntropyLoss()?

时间:2019-11-26 00:26:08

标签: python tensorflow keras pytorch

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

我快完成了,但是最后一个问题是耽误了我的工作。

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

感谢您的帮助。

1 个答案:

答案 0 :(得分:2)

根据标签和输出的类型,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