Merge pull request #42 from mihael147work/master

Small fix
risenW · Apr 6, 2020 · 2935fc6 · 2935fc6
2 parents eca0b62 + 5bcb259
commit 2935fc6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datasist/model.py b/datasist/model.py
@@ -257,7 +257,7 @@ def get_regression_report(y_true=None, prediction=None, show_r2_plot=True, save_
     '''
     mae = mean_absolute_error(y_true, prediction)
     mse = mean_squared_error(y_true, prediction)
-    msle = precision_score(y_true, prediction)
+    msle = mean_squared_log_error(y_true, prediction)
     r2 = r2_score(y_true, prediction)
 
     print("Mean Absolute Error: ", round(mae, 5))