Table 5. Train_RMSE, Train_R2, Test_RMSE, and Test_R2 of Various Machine Learning Models.
| model | Train_RMSE | Train_R2 | Test_RMSE | Test_R2 |
|---|---|---|---|---|
| GBDT | 0.231 | 0.992 | 0.001 | 1.000 |
| lasso | 2.368 | 0.169 | 0.956 | 0.204 |
| KNN | 2.378 | 0.162 | 0.771 | 0.483 |
| model | Train_RMSE | Train_R2 | Test_RMSE | Test_R2 |
|---|---|---|---|---|
| GBDT | 0.231 | 0.992 | 0.001 | 1.000 |
| lasso | 2.368 | 0.169 | 0.956 | 0.204 |
| KNN | 2.378 | 0.162 | 0.771 | 0.483 |