Nvidia DIGITS - 超过预测的前 5 名 类
Nvidia DIGITS - more than top 5 predicted classes
我正在使用 nvidia-digits 训练 CNN 并获得预测。但是默认情况下 .../classify_one.json
returns 前 5 个预测。我需要一个更大的 n
,如何修改我的请求参数以获得顶部 n
例如?
我在 DIGITS 文档中找不到任何内容。
尝试修改模型 prototxt
文件的顶 "Accuracy"
层的 accuracy_param
。你应该有类似
的东西
accuracy_param {
top_k: 5
}
将 top_k: 5
更改为您希望获得的数字,例如 top_k: 10
以获得前 10 名 类。
我正在使用 nvidia-digits 训练 CNN 并获得预测。但是默认情况下 .../classify_one.json
returns 前 5 个预测。我需要一个更大的 n
,如何修改我的请求参数以获得顶部 n
例如?
我在 DIGITS 文档中找不到任何内容。
尝试修改模型 prototxt
文件的顶 "Accuracy"
层的 accuracy_param
。你应该有类似
accuracy_param {
top_k: 5
}
将 top_k: 5
更改为您希望获得的数字,例如 top_k: 10
以获得前 10 名 类。