From 437429143181bc1ca1a616dc41842db011027f67 Mon Sep 17 00:00:00 2001 From: Willi Richert Date: Tue, 22 Mar 2016 22:06:15 +0100 Subject: [PATCH] Show winning classes --- contrib/Python/cntk/examples/LogReg/logreg.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/contrib/Python/cntk/examples/LogReg/logreg.py b/contrib/Python/cntk/examples/LogReg/logreg.py index 7f9919afb61f..9f32c8cde3d8 100644 --- 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))