c++ - 计算 Caffe 中的前 5 错误率？

Question

我使用 synset 从 softmax 输出计算排序前 k 预测。

这给了我前 5 个类名。但我想知道如何计算它的百分比。我的意思是前 5% 的错误。

谁能指导一下。谢谢。

Answer 1

在训练期间，您可以使用精度层的“top_k”参数计算“top k”精度。这是修改 bvlc_reference_caffe's train_val.prototxt 中现有精度层的示例:

layer {
  name: "accuracy_top_5"
  type: "Accuracy"
  bottom: "fc8"
  bottom: "label"
  top: "accuracy_top_5"
  accuracy_param {
    top_k: 5
  }
  include {
    phase: TEST
  }
}