网站首页 文章专栏 20190920样本不均衡训练.md
20190920样本不均衡训练.md
创建于:2021-07-04 08:29:06 更新于:2024-05-06 15:20:23 羽瀚尘 315

方法1:
网络输出为64x55的矩阵,激活后为loss_sigmoid
class_weights为长度为55的列表,代表各种类别在最终的loss中所占比重

loss_simoid 矩阵乘法 class_weights
可以实现目的,但是矩阵乘法有乘加的过程,导致最终loss大了55倍。

self.loss_sigmoid = tf.nn.sigmoid_cross_entropy_with_logits(logits=logits, labels=self.y)
self.loss_1 = tf.matmul(self.loss_sigmoid, tf.reshape(tf.to_float(self.class_weights), [55,1]))
self.loss_2 = tf.reduce_mean(self.loss_1, name="loss")

错误方法1:
这里是一个矩阵数乘另一个矩阵,结果是第二个矩阵的沿拓,千万注意
self.loss_1 = tf.multiply(tf.to_float(self.class_weights), self.loss_sigmoid)