Added new file of model comparison

0fd60e93 · IT19954806 Devindi L.A.P.S · adbdc6da · 0fd60e93
Commit 0fd60e93 authored Feb 17, 2023 by IT19954806 Devindi L.A.P.S
Hide whitespace changes
Inline Side-by-side

Showing with 11 additions and 0 deletions

Model Comparison Model Comparison +11 -0

No files found.
--- a/Model Comparison
+++ b/Model Comparison
+linreg_stats = [linreg_rmse, linreg_mae, linreg_r2]
+rf_stats = [rf_rmse, rf_mae, rf_r2]
+xgb_stats = [xgb_rmse, xgb_mae, xgb_r2]
+plt.figure(figsize=(15,7))
+plt.plot(linreg_stats)
+plt.plot(rf_stats)
+plt.plot(xgb_stats)
+plt.title("Model Comparison between Linear Regression, Random Forest and XG Boost")
+plt.xticks([0,1,2], labels=['RMSE','MAE','R2 Score'])
+plt.legend(["Linear Regression", "Random Forest", "XG Boost"])
+plt.show()
\ No newline at end of file