Tensorflow类平衡用于训练多类分类网络

时间:2016-04-23 23:52:30

标签: neural-network tensorflow image-segmentation conv-neural-network

我正在为多个班级培训分类神经网络。我有非常不平衡的课程(大约80:10:5:5比例)。我想在损失函数中使用某种权重平衡来防止神经网络过度预测大多数类。有人知道如何在Tensorflow中进行类平衡吗?

P.S。我无法通过对少数类进行过度采样来解决这个问题,因为我正在训练一个卷积 - 反卷积神经网络来进行医学图像分割。在此任务中,每个像素都分配给一个不同的类。我无法在此任务中过度采样像素。

非常感谢!:D

0 个答案:

没有答案
相关问题