Implement custom loss function in Tensorflow 2.0
问题 I'm building a model for Time series classification. The data is very unbalanced so I've decided to use a weighted cross entropy function as my loss. Tensorflow provides tf.nn.weighted_cross_entropy_with_logits but I'm not sure how to use it in TF 2.0. Because my model is build using tf.keras API I was thinking about creating my custom loss function like this: pos_weight=10 def weighted_cross_entropy_with_logits(y_true,y_pred): return tf.nn.weighted_cross_entropy_with_logits(y_true,y_pred,pos