如何在keras模型中使用张量流损失函数？

Question

我正在使用Tensorflow 1.14，并且已经使用Keras设计了一个模型。我想在编译模型时使用remotes 2.1.0 2019-06-24 [1] CRAN (R 3.6.1)，但在Keras中看不到任何等效的损失函数。有什么办法可以在我的模型中使用它吗？

我当前要编译的代码：

tf.nn.sparse_softmax_cross_entropy_with_logits

谢谢您的帮助。

Answer 1

您可以将tf.losses.sparse_categorical_crossentropy设置为from_logits的{​​{1}}并将其包装在函数中

True

但是，如果您坚持使用import tensorflow as tf def my_tf_loss_fn(y_true, y_pred): return tf.losses.sparse_categorical_crossentropy(y_true, y_pred, from_logits=True) model = tf.keras.applications.ResNet50() model.compile(loss=my_tf_loss_fn, optimizer='adam') 我想不出一种干净的方法来做到这一点，但这有效

tf.nn.sparse_softmax_cross_entropy_with_logits