How to get top-5 or top-3 accuracy by ConfusionMatrix? #140

Activity

1. taineleau-zz commented on Oct 13, 2016

   

   taineleau-zz

   on Oct 13, 2016

   Last edited by taineleau-zz

   More actions

   hi,

   I think you need to hack the ConfusionMatrix.lua file if you want to get the top-k acc.

   And there're two ways to calculate top-k acc., hope the following helps.

   1. please kindly check the resnet training script L107.
   2. Torchnet's utils for topk.