Data Analysis for the AirQuality

Author: AbuBakkar32

UniversityProject/air_quality_full_final_with_lags.py

@@ -87,7 +87,7 @@
 from sklearn.linear_model import LinearRegression, Ridge, Lasso, ElasticNet
 from sklearn.ensemble import RandomForestRegressor, GradientBoostingRegressor
 
-from sklearn.metrics import r2_score, mean_absolute_error, mean_squared_error
+from sklearn.metrics import r2_score, mean_absolute_error, mean_squared_error, root_mean_squared_error
 from sklearn.inspection import permutation_importance, partial_dependence
 
 try:
@@ -892,7 +892,7 @@ def fit_best_holdout(
 
     r2 = r2_score(y_test, y_pred)
     mae = mean_absolute_error(y_test, y_pred)
-    rmse = mean_squared_error(y_test, y_pred, squared=False)
+    rmse = root_mean_squared_error(y_test, y_pred)
 
     print(f"HOLDOUT R2:   {r2:.4f}")
     print(f"HOLDOUT MAE:  {mae:.4f}")