Skip to main content

Table 1 Overview of the machine learning models employed in our study. The first column refers to the model name, the second column to how class imbalance was dealt with, the third column to the ML algorithm used, the fourth column to whether hyperparameter tuning (HT) was performed, and the fifth column to whether environmental socioeconomic data (ESED) were included in the analysis

From: Identification of risk factors for involuntary psychiatric hospitalization: using environmental socioeconomic data and methods of machine learning to improve prediction

Model Name Class Imbalance Algorithm Hyperparameter
Tuning (HT)
ESED
Model 1 Weighting CHAID
Model 2 Imputation CHAID
Model 3 Imputation CART
Model 4 Imputation CART
Model 5 Imputation CART