From e5990b3dafa2b95fae7b7bfaac4dcd309a20d151 Mon Sep 17 00:00:00 2001
From: Brian Chu <bc@brianchu.com>
Date: Tue, 13 Oct 2015 03:50:53 -0700
Subject: [PATCH] In 00-classification example, get correct class label index

---
 examples/00-classification.ipynb | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/00-classification.ipynb b/examples/00-classification.ipynb
index 46bbb19..89b7dd3 100644
--- a/examples/00-classification.ipynb
+++ b/examples/00-classification.ipynb
@@ -119,7 +119,7 @@
    "source": [
     "net.blobs['data'].data[...] = transformer.preprocess('data', caffe.io.load_image(caffe_root + 'examples/images/cat.jpg'))\n",
     "out = net.forward()\n",
-    "print(\"Predicted class is #{}.\".format(out['prob'].argmax()))"
+    "print(\"Predicted class is #{}.\".format(out['prob'][0].argmax()))"
    ]
   },
   {
-- 
2.7.4