From 1256c2ce6968affab9a0d6f71bcee0ecd74241e1 Mon Sep 17 00:00:00 2001
From: QuantumNovice <43876848+QuantumNovice@users.noreply.github.com>
Date: Sat, 20 Jul 2019 18:14:09 +0500
Subject: [PATCH] Update random_forest_regression.py

---
 .../random_forest_regression/random_forest_regression.py        | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/machine_learning/random_forest_regression/random_forest_regression.py b/machine_learning/random_forest_regression/random_forest_regression.py
index b984b02b6e6e..85ce0676b598 100644
--- a/machine_learning/random_forest_regression/random_forest_regression.py
+++ b/machine_learning/random_forest_regression/random_forest_regression.py
@@ -30,7 +30,7 @@
 regressor.fit(X, y)
 
 # Predicting a new result
-y_pred = regressor.predict(6.5).reshape(1, -1)
+y_pred = regressor.predict([[6.5]])
 
 # Visualising the Random Forest Regression results (higher resolution)
 X_grid = np.arange(min(X), max(X), 0.01)