Nvidia DIGITS — более 5 лучших прогнозируемых классов

Я использую nvidia-digits, чтобы тренируйте CNN и получайте прогнозы. Однако .../classify_one.json по умолчанию возвращает 5 лучших прогнозов. Мне нужно большее n, как я могу изменить параметр моего запроса, чтобы, например, получить верхнее n?

В документации DIGITS ничего не нашел.


deep-learning neural-network caffe nvidia-digits
person Santino    schedule 13.06.2016    source источник

Ответы (1)


arrow_upward
2
arrow_downward

Попробуйте изменить accuracy_param верхнего "Accuracy" слоя prototxt файла модели. У вас должно быть что-то вроде

accuracy_param {
  top_k: 5
}

Измените top_k: 5 на число, которое вы хотите получить, например, top_k: 10 для получения 10 лучших классов.

person Shai    schedule 14.06.2016