Skip to main content

Table 2 Hyperparameters

From: Development and validation of a novel prediction model to identify patients in need of specialized trauma care during field triage: design and rationale of the GOAT study

Parameter

Explanation

Free

 Learning rate

Shrinkage rate (how much will the weights be adjusted every iteration).

 Number of leaves

Maximum number of leaves in one tree.

 Lambda L1

L1 regularization.

 Lambda L2

L2 regularization.

 Feature fraction

Randomly select part of the predictors on each iteration.

Fixed

 Early stopping

The cross-validation score needs to improve at least every n round to continue with the next boosting iteration.

 Maximum depth

Maximum tree depth (note that it is less relevant here since the tree grows leaf-wise).

 Minimum data

Minimal number of records in one leaf. A higher number prevents overfitting.

 Bagging fraction

Randomly select part of the data without resampling.

 Bagging frequency

Per how many rounds should bagging be applied.

 Unbalanced data

Does data need to be balanced or not.