Как я могу получить точность Top-k на PyCaffe на этапе обучения? - PullRequest
1 голос
/ 20 июня 2019

Я хотел бы знать, есть ли способ получить ошибку top-k на PyCaffe во время фазы обучения.

Я знаю, что есть параметр top_k для файла .prototxt, но есть ли способ, которым я могуможно использовать его на PyCaffe?

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

1 Ответ

0 голосов
/ 26 июня 2019

Для всех, кто интересуется, я просто узнаю, что вам нужно нанести несколько слоев точности с разными числами top-k. Вот пример точности топ-3.

layer {
    name: "accuracy1"
    type: "Accuracy"
    bottom: "score"
    bottom: "label"
    top: "accuracy1"
    include {
      phase: TEST
    }
  }
  layer {
    name: "accuracy2"
    type: "Accuracy"
    bottom: "score"
    bottom: "label"
    top: "accuracy2"
    accuracy_param {
        top_k: 2
    }
    include {
      phase: TEST
    }
  }
  layer {
    name: "accuracy3"
    type: "Accuracy"
    bottom: "score"
    bottom: "label"
    top: "accuracy3"
    accuracy_param {
        top_k: 3
    }
    include {
      phase: TEST
    }
  }
...