Show winning classes

naiduv · Mar 22, 2016 · 4374291 · 4374291
1 parent 21fbaa2
commit 4374291
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/contrib/Python/cntk/examples/LogReg/logreg.py b/contrib/Python/cntk/examples/LogReg/logreg.py
@@ -26,4 +26,4 @@
         ctx.train()
 
         result = ctx.eval(out)
-        print(result[:3])
+        print(result.argmax(axis=1))