Table 4 Three evaluation indicators on the training set and the test set (FS).
Model | Train set | Test set | ||||
|---|---|---|---|---|---|---|
R2 | RMSE | MAE | R2 | RMSE | MAE | |
LR | 0.82 | 3.12 | 2.27 | 0.92 | 2.21 | 1.73 |
KNN | 0.85 | 2.83 | 2.00 | 0.90 | 2.48 | 1.82 |
RF | 0.97 | 1.16 | 0.80 | 0.91 | 2.32 | 1.62 |
XGBoost | 0.97 | 1.33 | 0.87 | 0.91 | 2.32 | 1.73 |