dice_ml.utils.sample_architecture package
Submodules
dice_ml.utils.sample_architecture.vae_model module
- class dice_ml.utils.sample_architecture.vae_model.AutoEncoder(*args: Any, **kwargs: Any)[source]
Bases:
torch.nn.Module
Bases: torch.nn.Module