DeepCTR icon indicating copy to clipboard operation
DeepCTR copied to clipboard

MTL多目标损失函数权重设置

Open wushupei1997 opened this issue 4 years ago • 1 comments

MTL中,多个任务的损失函数默认是按照权重为1进行求和得到最终总的损失,想问下如何给不同的任务的损失设置不同的权重,保证不同任务的损失尺度是相同的

wushupei1997 avatar Oct 14 '21 13:10 wushupei1997

keras.model.compile 有一个 loss_weights 参数,或者自己写一个 loss 函数的实现

cbjs avatar Oct 15 '21 07:10 cbjs