Table 2 Hyperparameter ranges for successive halving.

From: Separation of organic molecules from water by design of membrane using mass transfer model analysis and computational machine learning

Model

Hyperparameter

Range

DT

max_depth

[5, 10, 15, 20, None]

RF

min_samples_split

[2, 5, 10]

n_estimators

[50, 100, 200]

ET

max_features

[‘auto’, ‘sqrt’, ‘log2’]

n_estimators

[50, 100, 200]

min_samples_leaf

[1, 2, 4]

HBGB

max_leaf_nodes

[15, 31, 63]

learning_rate

[0.01, 0.05, 0.1, 0.2]

max_depth

[3, 5, 7, 10]

Back to article page