TABLE 4.
Algorithm performance results (test set).
| Algorithm | R2 | MSE | MAE | MedAE | RE |
|---|---|---|---|---|---|
| XGBoost | 0.37 | 4.35 | 1.65 | 1.33 | 0.22 |
| RF | 0.34 | 4.57 | 1.68 | 1.40 | 0.24 |
| Extra-Trees | 0.39 | 4.17 | 1.58 | 1.22 | 0.22 |
| GBDT | 0.37 | 4.30 | 1.61 | 1.34 | 0.22 |
| AdaBoost | 0.38 | 4.25 | 1.63 | 1.30 | 0.23 |
| Lasso | 0.42 | 3.98 | 1.51 | 1.13 | 0.21 |
XGBoost: eXtreme Gradient Boosting, RF: Random Forest, GBDT: Gradient Boosting Decision Tree, AdaBoost: Adaptive boosting, MSE: Mean Square Error, MAE: Mean Absolute Error, MedAE: Median Absolute Error, RE: Root Mean Square Error.