Skip to main content
. 2021 Jul 8;4:693674. doi: 10.3389/fdata.2021.693674

TABLE 5.

An overview of all imputation methods and their hyperparameters we optimized. Mean/mode imputation does not have any hyperparameters, and Discriminative DL is optimized using autokeras, which is why we do not explicitly define a hyperparameter grid.

Imputation method Hyperparameters Grid size
Name Values
Mean/mode
k-NN n_neighbors (1, 3, 5) 3
 Random forest n_estimators (10, 50, 100) 3
Discriminative DL a
 VAE n_hidden_layers (0, 1, 2) 3
 GAIN alpha (1, 10) 16
hint_rate (0.7, 0.9)
generator_learning_rate (0.0001, 0.0005)
discriminator_learning_rate (0.00001, 0.00005)
a

Optimized using autokeras, see Section 3.4.4.