【发布时间】:2016-10-14 05:07:52
【问题描述】:
我正在使用nvidia-digits 来训练 CNN 并获得预测。然而,.../classify_one.json 默认返回前 5 个预测。我需要更大的n,例如如何修改我的请求参数以获得顶部n?
我在 DIGITS 文档中找不到任何内容。
【问题讨论】:
标签: nvidia-digits neural-network deep-learning caffe nvidia-digits
我正在使用nvidia-digits 来训练 CNN 并获得预测。然而,.../classify_one.json 默认返回前 5 个预测。我需要更大的n,例如如何修改我的请求参数以获得顶部n?
我在 DIGITS 文档中找不到任何内容。
【问题讨论】:
标签: nvidia-digits neural-network deep-learning caffe nvidia-digits
尝试修改模型prototxt 文件的顶层"Accuracy" 的accuracy_param。你应该有类似的东西
accuracy_param {
top_k: 5
}
将 top_k: 5 更改为您希望获得的数字,例如,top_k: 10 用于获得前 10 个班级。
【讨论】: