Research Article

Interpretable Machine Learning to Optimize Early In-Hospital Mortality Prediction for Elderly Patients with Sepsis: A Discovery Study

Table 1

Hyperparameters and tuning strategies for each model in the study.

ModelHyperparametersGrid-search setting

K-nearest neighborsNumber of neighbors5, 10
Neighbor weightUniform, distance
Decision treeMaximum depth10, 50, 100
Random forestNumber of estimators10, 50, 100
Maximum depth10, 50, 100
Logistic regressionCs0.1, 1, 10, 100, 1000
LGBMLearning rate0.01, 0.1, 1
Number leaves5, 10, 15, 20
Maximum depth3, 5, 7, 9, 11
XGBoostLearning rate0.01, 0.1, 1
Number leaves5, 10, 15, 20
Maximum depth3, 5, 7, 9, 11