在我的项目中,负面实例远远超过正面实例,因此我想给出具有更大权重的正面实例。 我的目标是:
loss = 0.0
if y_label==1:loss += 100 * cross_entropy
else:loss += cross_entropy
如何在张量流[?]
中实现这一点答案 0 :(得分:1)
让losses
成为批次中示例的损失值的向量(等级-1张量)。让y
成为相应标签的向量。然后,您可以通过
weights = w_pos*y + w_neg*(1.0-y)
loss = tf.reduce_mean(weights*losses)
此处,w_pos
和w_neg
是常量标量值(示例中为w_pos=100.0
和w_neg=1.0
)。然后,向量weights
的值为w_pos
,其中标签等于1,w_neg
为等于0.然后将weights
元素与{{1}相乘根据相应的标签权衡losses
中的值,然后取均值。