Algorithm | R2 (training fold) | MAE (training fold) | R2 (cross-validation fold) | MAE (cross-validation fold) |
---|---|---|---|---|
Linear regression | 0.72 | 418 | 0.50 | 531 |
Ridge regression | 0.68 | 443 | 0.56 | 509 |
LASSO regression | 0.69 | 427 | 0.59 | 488 |
Elastic net | 0.68 | 439 | 0.58 | 500 |
Random forest | 0.86 | 294 | 0.58 | 484 |
GBDT | 0.93 | 198 | 0.62 | 459 |