From 2719bca726fe96b89120d547d0daa908a0fd5b46 Mon Sep 17 00:00:00 2001 From: Maxim Smolskiy Date: Tue, 15 Nov 2022 22:05:21 +0300 Subject: [PATCH] Fix FutureWarning in ensemble/plot_gradient_boosting_regression.html --- examples/ensemble/plot_gradient_boosting_regression.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/ensemble/plot_gradient_boosting_regression.py b/examples/ensemble/plot_gradient_boosting_regression.py index dc29bfbda8f77..3e378e8af7203 100644 --- a/examples/ensemble/plot_gradient_boosting_regression.py +++ b/examples/ensemble/plot_gradient_boosting_regression.py @@ -94,7 +94,7 @@ test_score = np.zeros((params["n_estimators"],), dtype=np.float64) for i, y_pred in enumerate(reg.staged_predict(X_test)): - test_score[i] = reg.loss_(y_test, y_pred) + test_score[i] = mean_squared_error(y_test, y_pred) fig = plt.figure(figsize=(6, 6)) plt.subplot(1, 1, 1)