Skip to main content
. 2021 May 6;78:189–202. doi: 10.1016/j.jsr.2021.04.007

Table 8.

Examined and optimized hyperparameters for harsh braking/100 km XGBoost algorithms.

Hyperparameter Examined range Optimized Value
Learning rate 0.001–0.6 0.374
Gamma 0.001–10 1.37
Maximum tree depth 2–10 6
Minimum child weight 1–10 1
Number of rounds 1–250 242
Mean Squared Error as low as possible 0.018