TABLE 1 |.
Parameter | Description* | Values | Best parameters | ||||
---|---|---|---|---|---|---|---|
Pressure | Volume | Myofiber stress | Trans-myofiber stress | Shear stress | |||
learning_rate | The learning rate of the boosting | 0.01, 0.05, 0.1 | 0.05 | 0.05 | 0.05 | 0.01 | 0.1 |
n_estimators | Number of trees in the model | 500, 1,000, 1,500 | 500 | 1,500 | 1,500 | 1,000 | 1,500 |
Max_depth | Maximum depth of trees | 7,15,20 | 15 | 7 | 7 | 7 | 7 |
The number of folds in cross validation (cv) = 3.
XGBoost python package guide.