Merge pull request #1 from cseas/patch-1

Fixed issue random-forests#4
cseas · Jul 15, 2018 · c480c41 · c480c41
2 parents d645175 + 693f6eb
commit c480c41
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ep7.ipynb b/ep7.ipynb
@@ -294,7 +294,8 @@
    ],
    "source": [
     "# here's one it gets right\n",
-    "print (\"Predicted %d, Label: %d\" % (classifier.predict(test_data[0]), test_labels[0]))\n",
+    "prediction = classifier.predict(np.array([test_data[0]], dtype=float), as_iterable=False)",
+    "print (\"Predicted %d, Label: %d\" % (prediction, test_labels[0]))\n",
     "display(0)"
    ]
   },
@@ -325,7 +326,8 @@
    ],
    "source": [
     "# and one it gets wrong\n",
-    "print (\"Predicted %d, Label: %d\" % (classifier.predict(test_data[8]), test_labels[8]))\n",
+    "prediction = classifier.predict(np.array([test_data[0]], dtype=float), as_iterable=False)",
+    "print (\"Predicted %d, Label: %d\" % (prediction, test_labels[8]))\n",
     "display(8)"
    ]
   },