fixed error on stock prediction tutorial

x4nth055 · x4nth055 · commit 637fb0e5f878 · 2020-06-01T16:38:15.000+02:00
diff --git a/machine-learning/stock-prediction/test.py b/machine-learning/stock-prediction/test.py
@@ -58,7 +58,7 @@ def predict(model, data, classification=False):
 # evaluate the model
 mse, mae = model.evaluate(data["X_test"], data["y_test"], verbose=0)
 # calculate the mean absolute error (inverse scaling)
-mean_absolute_error = data["column_scaler"]["adjclose"].inverse_transform(mae.reshape(1, -1))[0][0]
+mean_absolute_error = data["column_scaler"]["adjclose"].inverse_transform([[mae]])[0][0]
 print("Mean Absolute Error:", mean_absolute_error)
 # predict the future price
 future_price = predict(model, data)
