Table 6 Three evaluation indicators on the training set and the test set (TS).
Model | Train set | Test set | ||||
|---|---|---|---|---|---|---|
R2 | RMSE | MAE | R2 | RMSE | MAE | |
LR | 0.71 | 1.46 | 1.06 | 0.63 | 1.73 | 1.25 |
KNN | 0.89 | 0.89 | 0.60 | 0.75 | 1.41 | 0.91 |
RF | 0.96 | 0.56 | 0.38 | 0.84 | 1.13 | 0.74 |
XGB | 0.97 | 0.44 | 0.32 | 0.87 | 1.04 | 0.68 |