相关疑难解决方法(0)

使用混淆矩阵理解多标签分类器

我有一个包含 12 个类的多标签分类问题。我使用slimTensorflow训练使用预训练的模型模型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

8
推荐指数
1
解决办法
8826
查看次数