From 3ffa1b57b9cc502ef71755c9e6938831e07df3b4 Mon Sep 17 00:00:00 2001 From: paillarj Date: Tue, 15 Oct 2024 20:53:49 +0200 Subject: [PATCH] deprecated loss --- examples/plot_diabetes_variable_importance_example.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/plot_diabetes_variable_importance_example.py b/examples/plot_diabetes_variable_importance_example.py index f753792..f5126e2 100644 --- a/examples/plot_diabetes_variable_importance_example.py +++ b/examples/plot_diabetes_variable_importance_example.py @@ -101,7 +101,7 @@ score = r2_score( y_true=y[test_index], y_pred=regressor_list[i].predict(X[test_index]) ) - mse = mean_squared_error( + mse = root_mean_squared_error( y_true=y[test_index], y_pred=regressor_list[i].predict(X[test_index]) )