Table 8 Three evaluation indicators on the training set and the test set.
Model | Train set | Test set | ||||
|---|---|---|---|---|---|---|
R2 | RMSE | MAE | R2 | RMSE | MAE | |
LR | 0.63 | 1.62 | 1.24 | 0.61 | 1.54 | 1.16 |
KNN | 0.85 | 1.05 | 0.75 | 0.70 | 1.34 | 1.03 |
RF | 0.92 | 0.76 | 0.56 | 0.72 | 1.28 | 0.94 |
XGBoost | 0.95 | 0.57 | 0.42 | 0.80 | 1.10 | 0.84 |