CEVAE Model Settings

[Leo-T-Zang]

Hi,

Does anyone know what following parameters mean for CEVAE?
The link of notebook is https://github.com/uber/causalml/blob/master/examples/cevae_example.ipynb
Specifically, if I have a dataset with high dimension, say 700 features, which one of dimension I should adjust?

# cevae model settings

latent_dim = 20
hidden_dim = 200

Thanks