2

I am using to train a CNN and get predictions. However the .../classify_one.json returns top 5 predictions by default. I need a larger n, how can I modify my request param to get the top n for example?

I couldn't find anything in DIGITS documentation.

Shai
  • 111,146
  • 38
  • 238
  • 371
Santino
  • 776
  • 2
  • 11
  • 29

1 Answers1

2

Try modifying the accuracy_param of your top "Accuracy" layer of the model's prototxt file. You should have something like

accuracy_param {
  top_k: 5
}

Change the top_k: 5 to the number you wish to get, e.g., top_k: 10 for getting top 10 classes.

Shai
  • 111,146
  • 38
  • 238
  • 371