sales-forecast-model / experiment_logs.csv
Abhik19's picture
Upload folder using huggingface_hub
834eda1 verified
model_name,parameters,train_rmse,test_rmse,train_mae,test_mae,train_r2,test_r2,training_time
Decision Tree,"{'max_depth': 10, 'min_samples_split': 5, 'min_samples_leaf': 2, 'random_state': 42}",224.39530290081427,320.2862156093461,103.61202500299142,144.07551596074504,0.9555983787745922,0.91009512501219,0.038845062255859375
Bagging,"{'n_estimators': 50, 'random_state': 42, 'n_jobs': -1}",108.50256998441432,287.8643679691224,41.541703252496426,111.69428648031945,0.9896187092677091,0.9273755983331324,5.753796339035034
Random Forest,"{'n_estimators': 100, 'max_depth': 15, 'min_samples_split': 5, 'min_samples_leaf': 2, 'random_state': 42, 'n_jobs': -1}",179.95402471933053,280.1628522465805,65.176864288256,107.0770687953505,0.9714441933786859,0.9312095982549087,2.94158935546875
AdaBoost,"{'n_estimators': 100, 'learning_rate': 0.1, 'random_state': 42}",474.52428811286796,472.09692161063026,369.9771454286811,369.1592529361528,0.8014416311336914,0.80466994041514,1.2030410766601562
Gradient Boosting,"{'n_estimators': 100, 'learning_rate': 0.1, 'max_depth': 5, 'min_samples_split': 5, 'random_state': 42}",226.1356476787997,282.68403078012483,94.28165699668868,120.42179372377107,0.9549069758220278,0.9299659412754018,1.659700870513916
XGBoost (Tuned),"{'learning_rate': 0.05, 'max_depth': 7, 'n_estimators': 100}",212.67216238879408,282.08864546811253,82.64765498524011,115.13552294699042,0.9601165611139318,0.9302606401920417,0.254439115524292