Skip to main content

Table 2 Tuned parameters using grid-search

From: Predicting perinatal mortality based on maternal health status and health insurance service using homogeneous ensemble machine learning methods

Algorithm

Parameter

Values

Gradient boost

Criterion

friedman_mse

max_depth

15

n_estimators

100

Random_state

42

Random forest

Criterion

Entropy

max_features

sqrt

min_samples_split

3

n_estimators

200

random_state

0

max_depth

20

max_leaf_nodes

400

n_jobs

− 1

Cat boost

Verbose

False

eval_metric

AUC

Iterations

500

thread_count

None

random_state

1.0