deprecated loss

mind-inria · Oct 15, 2024 · 3ffa1b5 · 3ffa1b5
1 parent 50ca014
commit 3ffa1b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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])
     )