小编use*_*101的帖子

我正在构建时间序列分类模型。数据非常不平衡，所以我决定使用加权交叉熵函数作为我的损失。

Tensorflow 提供了tf.nn.weighted_cross_entropy_with_logits，但我不确定如何在 TF 2.0 中使用它。因为我的模型是使用 tf.keras API 构建的，所以我正在考虑像这样创建我的自定义损失函数：

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_weight)

# .....
model.compile(loss=weighted_cross_entropy_with_logits,optimizer="adam",metrics=["acc"])

我的问题是：有没有办法直接将 tf.nn.weighted_cross_entropy_with_logits 与 tf.keras API 一起使用？