我有一个包含 12 个类的多标签分类问题。我使用slim的Tensorflow训练使用预训练的模型模型ImageNet。以下是训练和验证中每个类别的存在百分比
Training Validation
class0 44.4 25
class1 55.6 50
class2 50 25
class3 55.6 50
class4 44.4 50
class5 50 75
class6 50 75
class7 55.6 50
class8 88.9 50
class9 88.9 50
class10 50 25
class11 72.2 25
Run Code Online (Sandbox Code Playgroud)
问题是模型没有收敛,并且验证集上的ROC曲线 ( Az) 的下限很差,例如:
Az
class0 0.99
class1 0.44
class2 0.96
class3 0.9
class4 0.99
class5 0.01
class6 0.52
class7 0.65
class8 0.97
class9 0.82
class10 0.09
class11 0.5
Average …Run Code Online (Sandbox Code Playgroud) python machine-learning confusion-matrix deep-learning tensorflow