TensorFlow 模型建立与训练

谢谢, 我试试

1 Like

请问多任务学习应当如何定义Loss函数,不使用keras.model.fit 。 :grinning:

这就完全要看具体场景了,多任务学习的范围还是比较大的,可以参考一下相关论文里用了什么loss。

能不能举个简单例子说明,就用最简单的模型,有两个输入,两个输出,loss函数为两个binary_cross_entropy相加,权重设置为0.7,0.3,该如何实现呢?

那就按照你说的,

loss = 0.7 * tf.keras.losses.binary_crossentropy(y_true_1, y_pred_1) + \
       0.3 * tf.keras.losses.binary_crossentropy(y_true_2, y_pred_2)

是这个意思吗?