diff --git a/src/ageml/modelling.py b/src/ageml/modelling.py index cdafe2e..d59f9f3 100644 --- a/src/ageml/modelling.py +++ b/src/ageml/modelling.py @@ -291,7 +291,7 @@ def calculate_metrics(self, y_true, y_pred): MAE = metrics.mean_absolute_error(y_true, y_pred) rmse = metrics.mean_squared_error(y_true, y_pred, squared=False) r2 = metrics.r2_score(y_true, y_pred) - p, pval = stats.pearsonr(y_true, y_pred) + p, _ = stats.pearsonr(y_true, y_pred) return MAE, rmse, r2, p def summary_metrics(self, array):