Table 2.
10-Fold cross-validation results of different machine learning models.
| Model name | Mean MSE(SD) | Mean MAE(SD) | Mean R2(SD) |
|---|---|---|---|
| Linear regression (LR) | 7.00 (0.48) | 2.16 (0.07) | 0.69(0.02) |
| Decision tree (DT) | 1.67 (0.32) | 0.54(0.05) | 0.93(0.01) |
| Random forest (RF) | 1.44 (0.21) | 0.55(0.04) | 0.93(0.01) |
| Support vector regressor (SVR) | 1.62 (0.18) | 0.57(0.03) | 0.92(0.01) |
| Extreme gradient boosting (XGB) | 1.58 (0.21) | 0.68(0.04) | 0.93(0.01) |
| K-nearest neighbour (KNN) | 1.74 (0.30) | 0.63(0.04) | 0.92(0.01) |
| Artificial neural network (ANN) | 1.45 (0.19) | 0.59(0.04) | 0.93(0.01) |
The best value in each column is highlighted in bold.