Table 3 Optimal parameter values for 6 ML models
From: Improvement of APACHE II score system for disease severity based on XGBoost algorithm
Abbreviation | Hyperparameter | Value |
|---|---|---|
SVM | C | 2 |
Gamma | 1 | |
LR | C | 0.1 |
NB | Alpha | 1 |
ANN | Hidden_layer_sizes | 30 |
RF | n_estimators | 400 |
XGBoost | Lerning_rate | 0.1 |
n_estimator | 90 | |
Gamma | 0.05 | |
reg_lambda | 0.1 | |
Subsample | 0.6 | |
Min_child_weight | 1 | |
max_depth | 5 |