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

decoder(z)[source]
encoder(x)[source]
forward(x)[source]
normal_likelihood(x, mean, logvar, raxis=1)[source]
sample_latent_code(mean, logvar)[source]
class dice_ml.utils.sample_architecture.vae_model.CF_VAE(*args: Any, **kwargs: Any)[source]

Bases: torch.nn.Module

compute_elbo(x, c, pred_model)[source]
decoder(z)[source]
encoder(x)[source]
forward(x, c)[source]
normal_likelihood(x, mean, logvar, raxis=1)[source]
sample_latent_code(mean, logvar)[source]

Module contents