neural-network - Nvidia DIGITS - 超过前 5 个预测类别

Question

我正在使用nvidia-digits来训练 CNN 并获得预测。然而.../classify_one.json，默认情况下返回前 5 个预测。我需要一个更大的n，例如如何修改我的请求参数以获得顶部n？

我在 DIGITS 文档中找不到任何内容。

Answer 1

尝试修改模型文件accuracy_param的顶层。你应该有类似的东西"Accuracy"prototxt

accuracy_param {
  top_k: 5
}

将 更改为top_k: 5您希望获得的数字，例如top_k: 10获得前 10 类。