指定样本权重和损失权重时的tf.keras行为

时间:2019-10-15 20:33:13

标签: tensorflow keras

当您还在.compile()函数中指定损失权重时,使用model.predict_on_batch(x,y,w)时损失函数的描述是什么。

这是我的猜测: 假设您有2个损失作为损失函数的一部分,分别称为L1和L2。 L2逐个样本加权。我假设总体损失函数L为:

L = reduce_sum(L1_i + compileLossCoef * w_i * L2_i)

其中下标i表示批次中的特定样品,并使用.compile(loss = [L1,L2],loss_weights = [1,compileLossCoef])构建模型

这正确吗?

0 个答案:

没有答案