Parameters and Configurables

General and Default

model

• save_model : False

• seed_everything : True

• seed : 42

• z_dim : 5

• learning_rate : 0.001

• sparse : False

• threshold : 0

• return_mean : True

datamodule

• _target_ : multiviewae.base.dataloaders.MultiviewDataModule

• batch_size : None

• is_validate : True

• train_size : 0.9

• dataset : {‘_target_’: ‘multiviewae.base.datasets.MVDataset’}

encoder

• default

  • _target_ : multiviewae.architectures.mlp.Encoder

  • hidden_layer_dim : []

  • bias : True

  • non_linear : False

  • enc_dist

    • _target_ : multiviewae.base.distributions.Default

decoder

• default

  • _target_ : multiviewae.architectures.mlp.Decoder

  • hidden_layer_dim : []

  • bias : True

  • non_linear : False

  • dec_dist

    • _target_ : multiviewae.base.distributions.Default

prior

• _target_ : multiviewae.base.distributions.Normal

• loc : 0.0

• scale : 1.0

trainer

• _target_ : pytorch_lightning.Trainer

• accelerator : auto

• max_epochs : 10

• deterministic : False

• log_every_n_steps : 2

optimizer

• _target_ : torch.optim.Adam

callbacks

• model_checkpoint

  • _target_ : pytorch_lightning.callbacks.ModelCheckpoint

  • monitor : val_loss

  • mode : min

  • save_last : True

  • dirpath : ${out_dir}

• early_stopping

  • _target_ : pytorch_lightning.callbacks.EarlyStopping

  • monitor : val_loss

  • mode : min

  • patience : 299

  • min_delta : 0.001

  • verbose : True

logger

• _target_ : pytorch_lightning.loggers.tensorboard.TensorBoardLogger

• save_dir : ${out_dir}/logs

Model-specific

AE

DCCAE

• model

  • _lambda : 1

• optimizer

  • _target_ : torch.optim.LBFGS

DMVAE

• model

  • private : True

  • s_dim : 3

  • beta : 1

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • multiple_latents : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

DVCCA

• model

  • beta : 1

  • private : False

  • sparse : True

  • threshold : 0

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

JMVAE

• model

  • alpha : 1

  • warmup : 0

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

mAAE

• discriminator

  • _target_ : multiviewae.architectures.mlp.Discriminator

  • hidden_layer_dim : []

  • bias : True

  • non_linear : False

  • dropout_threshold : 0

mcVAE

• model

  • beta : 1

  • sparse : False

  • threshold : 0

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

me_mVAE

• model

  • beta : 1

  • join_type : PoE

  • warmup : 0

  • sparse : False

  • use_prior : False

  • weight_kld : False

  • weight_ll : False

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

mmJSD

• model

  • private : True

  • s_dim : 2

  • alpha : 1

  • beta : 1

  • weight_ll : False

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

mmVAE

• model

  • K : 1

  • beta : 1

  • DREG_loss : False

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

mmVAEPlus

• model

  • K : 1

  • beta : 1

  • u_dim : 3

  • w_dim : 2

  • z_dim : 5

  • learn_private_prior : True

  • learn_shared_prior : False

  • multiple_latents : True

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

MoPoEVAE

• model

  • beta : 1

  • weight_ll : False

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

mVAE

• model

  • beta : 1

  • join_type : PoE

  • sparse : False

  • threshold : 0.2

  • warmup : 10

  • use_prior : True

  • weight_ll : False

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

mvtCAE

• model

  • beta : 1

  • alpha : 0.5

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

mWAE

• discriminator

  • _target_ : multiviewae.architectures.mlp.Discriminator

  • hidden_layer_dim : []

  • bias : True

  • non_linear : False

  • dropout_threshold : 0

weighted_DMVAE

• model

  • private : True

  • s_dim : 3

  • beta : 1

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal

weighted_mVAE

• model

  • beta : 1

  • _lambda : 1

  • private : False

• encoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalEncoder

    • non_linear : False

    • enc_dist

      • _target_ : multiviewae.base.distributions.Normal

• decoder

  • default

    • _target_ : multiviewae.architectures.mlp.VariationalDecoder

    • non_linear : False

    • init_logvar : -3

    • dec_dist

      • _target_ : multiviewae.base.distributions.Normal