Fix for issue random-forests#4

cseas · Jul 15, 2018 · ddca55b · ddca55b
1 parent d645175
commit ddca55b
Showing 1 changed file with 2 additions and 1 deletion.
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)"
    ]
   },