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

✓

✓