Table 4 Optimal parameters of each model.
Model | NT | DT | Mtry | Splitting criterion | Learning rate | Loss function | Initial value |
---|---|---|---|---|---|---|---|
RF | 70 | 10 | 1 | MSE | - | - | - |
GBDT | 70 | 9 | 3 | Friedman MSE | 0.1 | MSE | Zero |
Model | NT | DT | Mtry | Splitting criterion | Learning rate | Loss function | Initial value |
---|---|---|---|---|---|---|---|
RF | 70 | 10 | 1 | MSE | - | - | - |
GBDT | 70 | 9 | 3 | Friedman MSE | 0.1 | MSE | Zero |