Index

A

all_subsets() (multivae.models.MoPoE method)

AutoConfig (class in multivae.models) AutoModel (class in multivae.models)

B

BaseConditionalDecoder (class in multivae.models.nn) BaseJointEncoder (class in multivae.models.nn) BaseMultilatentEncoder (class in multivae.models.nn) BaseMultiVAE (class in multivae.models)

BaseMultiVAEConfig (class in multivae.models) BaseSampler (class in multivae.samplers) BaseSamplerConfig (class in multivae.samplers) BaseTrainer (class in multivae.trainers) BaseTrainerConfig (class in multivae.trainers)

C

calc_joint_divergence() (multivae.models.MoPoE method) calculate_frechet_distance() (multivae.metrics.FIDEvaluator method) CelebAttr (class in multivae.data.datasets) check_and_set_posterior_blocks() (multivae.models.MHVAE method) check_input_dims() (multivae.models.BaseMultiVAE method) check_or_create_dataset() (multivae.data.datasets.TranslatedMMNIST method) Clustering (class in multivae.metrics) ClusteringConfig (class in multivae.metrics) CMVAE (class in multivae.models) CMVAEConfig (class in multivae.models) coherence_from_subset() (multivae.metrics.CoherenceEvaluator method) CoherenceEvaluator (class in multivae.metrics) CoherenceEvaluatorConfig (class in multivae.metrics) compute_all_conditional_fids() (multivae.metrics.FIDEvaluator method) compute_cond_nll() (multivae.models.BaseMultiVAE method) compute_fid_from_conditional_generation() (multivae.metrics.FIDEvaluator method) compute_joint_nll() (multivae.models.CMVAE method) (multivae.models.CRMVAE method) (multivae.models.DMVAE method) (multivae.models.MMVAE method) (multivae.models.MMVAEPlus method) (multivae.models.MoPoE method) (multivae.models.MVAE method) (multivae.models.MVTCAE method)

compute_joint_nll_paper() (multivae.models.MMVAE method) (multivae.models.MoPoE method) compute_k_lws() (multivae.models.MMVAE method) compute_mu_log_var_subset() (multivae.models.MVAE method) conditional_samples_subset() (multivae.metrics.Visualization method) CRMVAE (class in multivae.models) CRMVAEConfig (class in multivae.models) cross_coherences() (multivae.metrics.CoherenceEvaluator method) CUB (class in multivae.data.datasets) CVAE (class in multivae.models) CVAEConfig (class in multivae.models)

D

decode() (multivae.models.BaseMultiVAE method) (multivae.models.CVAE method) (multivae.models.Nexus method)

deterministic_mixture_component_selection() (multivae.models.MoPoE method) DMVAE (class in multivae.models) DMVAEConfig (class in multivae.models) dreg_looser() (multivae.models.MMVAE method)

E

encode() (multivae.models.BaseMultiVAE method) (multivae.models.CMVAE method) (multivae.models.CRMVAE method) (multivae.models.CVAE method) (multivae.models.DMVAE method) (multivae.models.JMVAE method) (multivae.models.JNF method) (multivae.models.MHVAE method) (multivae.models.MMVAE method) (multivae.models.MMVAEPlus method) (multivae.models.MoPoE method) (multivae.models.MVAE method) (multivae.models.MVTCAE method) (multivae.models.Nexus method) (multivae.models.TELBO method)

eval() (multivae.metrics.CoherenceEvaluator method) (multivae.metrics.Reconstruction method) eval_step() (multivae.trainers.BaseTrainer method)

F

FIDEvaluator (class in multivae.metrics) FIDEvaluatorConfig (class in multivae.metrics) fit() (multivae.samplers.BaseSampler method) (multivae.samplers.GaussianMixtureSampler method) (multivae.samplers.IAFSampler method) (multivae.samplers.MAFSampler method) forward() (multivae.models.BaseMultiVAE method) (multivae.models.CMVAE method) (multivae.models.CRMVAE method) (multivae.models.CVAE method) (multivae.models.DMVAE method) (multivae.models.JMVAE method) (multivae.models.JNF method) (multivae.models.MHVAE method) (multivae.models.MMVAE method) (multivae.models.MMVAEPlus method) (multivae.models.MoPoE method) (multivae.models.MVAE method) (multivae.models.MVTCAE method) (multivae.models.Nexus method) (multivae.models.nn.BaseConditionalDecoder method) (multivae.models.nn.BaseJointEncoder method) (multivae.models.nn.BaseMultilatentEncoder method) (multivae.models.TELBO method)

from_json_file() (multivae.models.AutoConfig class method)

G

GaussianMixtureSampler (class in multivae.samplers) GaussianMixtureSamplerConfig (class in multivae.samplers) generate_from_prior() (multivae.models.BaseMultiVAE method) (multivae.models.CMVAE method) (multivae.models.CVAE method) (multivae.models.DMVAE method) (multivae.models.MMVAE method) (multivae.models.MMVAEPlus method)

get_frechet_distance() (multivae.metrics.FIDEvaluator method)

I

IAFSampler (class in multivae.samplers) IAFSamplerConfig (class in multivae.samplers)

IncompleteDataset (class in multivae.data.datasets) inference() (multivae.models.MoPoE method) iwae_looser() (multivae.models.MMVAE method)

J

JMVAE (class in multivae.models) JMVAEConfig (class in multivae.models) JNF (class in multivae.models)

JNFConfig (class in multivae.models) joint_coherence() (multivae.metrics.CoherenceEvaluator method) joint_nll_from_subset() (multivae.metrics.LikelihoodsEvaluator method)

L

LikelihoodsEvaluator (class in multivae.metrics) LikelihoodsEvaluatorConfig (class in multivae.metrics) load_flows_from_folder() (multivae.samplers.IAFSampler method) (multivae.samplers.MAFSampler method)

load_from_folder() (multivae.models.AutoModel class method) load_from_hf_hub() (multivae.models.AutoModel class method) log_var_to_std() (multivae.models.MMVAE method)

M

MAFSampler (class in multivae.samplers) MAFSamplerConfig (class in multivae.samplers) MHD (class in multivae.data.datasets) MHVAE (class in multivae.models) MHVAEConfig (class in multivae.models) MMNISTDataset (class in multivae.data.datasets) MMVAE (class in multivae.models) MMVAEConfig (class in multivae.models) MMVAEPlus (class in multivae.models) MMVAEPlusConfig (class in multivae.models) MnistLabels (class in multivae.data.datasets) MnistSvhn (class in multivae.data.datasets) modality_encode() (multivae.models.MHVAE method) (multivae.models.MoPoE method) module multivae.data.datasets multivae.metrics multivae.metrics.base multivae.metrics.coherences multivae.metrics.fids multivae.metrics.latent_clustering multivae.metrics.likelihoods multivae.metrics.reconstruction multivae.metrics.visualization multivae.models.auto_model multivae.models.base multivae.models.cmvae multivae.models.crmvae multivae.models.cvae multivae.models.dmvae multivae.models.jmvae multivae.models.jnf multivae.models.mhvae multivae.models.mmvae multivae.models.mmvaePlus multivae.models.mopoe multivae.models.mvae multivae.models.mvtcae multivae.models.nexus multivae.models.nn.base_architectures multivae.models.telbo multivae.samplers multivae.samplers.base multivae.samplers.gaussian_mixture multivae.samplers.iaf_sampler multivae.samplers.maf_sampler multivae.trainers multivae.trainers.base multivae.trainers.multistage MoPoE (class in multivae.models) MoPoEConfig (class in multivae.models) MultimodalBaseDataset (class in multivae.data.datasets) MultistageTrainer (class in multivae.trainers) MultistageTrainerConfig (class in multivae.trainers) multivae.data.datasets module multivae.metrics module multivae.metrics.base module multivae.metrics.coherences module multivae.metrics.fids module

multivae.metrics.latent_clustering module multivae.metrics.likelihoods module multivae.metrics.reconstruction module multivae.metrics.visualization module multivae.models.auto_model module multivae.models.base module multivae.models.cmvae module multivae.models.crmvae module multivae.models.cvae module multivae.models.dmvae module multivae.models.jmvae module multivae.models.jnf module multivae.models.mhvae module multivae.models.mmvae module multivae.models.mmvaePlus module multivae.models.mopoe module multivae.models.mvae module multivae.models.mvtcae module multivae.models.nexus module multivae.models.nn.base_architectures module multivae.models.telbo module multivae.samplers module multivae.samplers.base module multivae.samplers.gaussian_mixture module multivae.samplers.iaf_sampler module multivae.samplers.maf_sampler module multivae.trainers module multivae.trainers.base module multivae.trainers.multistage module MVAE (class in multivae.models) MVAEConfig (class in multivae.models) MVTCAE (class in multivae.models) MVTCAEConfig (class in multivae.models)

N

Nexus (class in multivae.models)

NexusConfig (class in multivae.models)

P

pc_params (multivae.models.CMVAE property) predict() (multivae.models.BaseMultiVAE method) (multivae.models.CVAE method) (multivae.trainers.BaseTrainer method) predict_clusters() (multivae.models.CMVAE method)

prepare_train_step() (multivae.trainers.BaseTrainer method) (multivae.trainers.MultistageTrainer method) prepare_training() (multivae.trainers.BaseTrainer method) prune_clusters() (multivae.models.CMVAE method) pz_params (multivae.models.MMVAE property) (multivae.models.MMVAEPlus property)

R

random_mixture_component_selection() (multivae.models.MoPoE method) Reconstruction (class in multivae.metrics)

reconstruction_from_subset() (multivae.metrics.Reconstruction method) ReconstructionConfig (class in multivae.metrics) resume_training() (multivae.trainers.BaseTrainer method)

S

sample() (multivae.samplers.BaseSampler method) (multivae.samplers.GaussianMixtureSampler method) (multivae.samplers.IAFSampler method) (multivae.samplers.MAFSampler method) sanity_check() (multivae.models.BaseMultiVAE method) sanity_check_bottom_up() (multivae.models.MHVAE method) sanity_check_prior_blocks() (multivae.models.MHVAE method) sanity_check_top_down_blocks() (multivae.models.MHVAE method) save() (multivae.samplers.BaseSampler method) (multivae.samplers.IAFSampler method) (multivae.samplers.MAFSampler method)

save_checkpoint() (multivae.trainers.BaseTrainer method) save_model() (multivae.trainers.BaseTrainer method) set_bottom_up_blocks() (multivae.models.MHVAE method) set_decoders() (multivae.models.BaseMultiVAE method) set_decoders_dist() (multivae.models.BaseMultiVAE method) set_encoders() (multivae.models.BaseMultiVAE method) set_rescale_factors() (multivae.models.BaseMultiVAE method) set_subsets() (multivae.models.MoPoE method) set_top_down_blocks() (multivae.models.MHVAE method) subset_encode() (multivae.models.MHVAE method) subset_mask() (multivae.models.MoPoE method)

T

TELBO (class in multivae.models) TELBOConfig (class in multivae.models) train() (multivae.trainers.BaseTrainer method) train_step() (multivae.trainers.BaseTrainer method)

transform_for_plotting() (multivae.data.datasets.CUB method) (multivae.data.datasets.MnistLabels method) (multivae.data.datasets.MultimodalBaseDataset method) TranslatedMMNIST (class in multivae.data.datasets)

U

unconditional_fids() (multivae.metrics.FIDEvaluator method)

unconditional_samples() (multivae.metrics.Visualization method) update() (multivae.models.BaseMultiVAE method)

V

Visualization (class in multivae.metrics)

VisualizationConfig (class in multivae.metrics)