Table 4 Hyperparameter grid searched for the XGBoost model
From: Machine learning on multiple topological materials datasets
Parameter | Value | Parameter | Value |
|---|---|---|---|
Learning rate η | 0.23 | L2 regularization λ | 1.33 |
Maximal tree depth | [9, 10, 11] | Miniaml child weight | [0.1, 0.3, 0.5] |
Column subsampling by tree | [0.75, 0.78] | Column subsampling by node | [0.75, 0.78] |
