From: Brian Chu <bc@brianchu.com>
Date: Tue, 13 Oct 2015 10:50:53 +0000 (-0700)
Subject: In 00-classification example, get correct class label index
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e5990b3dafa2b95fae7b7bfaac4dcd309a20d151;p=platform%2Fupstream%2Fcaffe.git

In 00-classification example, get correct class label index
---

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()))"
    ]
   },
   {