################################################################################
#   CONFIGURATIONS FOR SWISS ROLL EXPERIMENT WITH VARIATIONAL CME PROCESS
################################################################################


# MODEL SPECIFICATIONS
model:
  # Name of model to use
  name: 'variational_cmp'

  # CME regularization parameter
  lbda: 0.0001

  # Number of inducing points
  n_inducing_points: 200

  # Whether to use noised individuals model or not
  use_individuals_noise: False

  # Random seed for inducing points selection
  seed: 42


# DATASET
dataset:
  # Number of points samples in swiss roll
  n_samples: 5000

  # Number of bags
  n_bags: 50

  # Variance of noise to apply on aggregate targets
  noise: 0.05

  # Random seed for dataset generation
  seed: 42



# TRAINING
training:
  # Name of model to use
  name: 'variational_cmp'

  # Whether to use noised individuals model or not
  use_individuals_noise: False

  # Learning rate
  lr: 0.05

  # Number of iterations
  n_epochs: 300

  # Weight of KL term in ELBO
  beta: 1

  # Seed for dataset shuffling
  seed: 42



# EVALUATION
evaluation:
  # Number of samples to use for NLL computation - doesn't scale
  chunk_size_nll: 2000

  # Random seed for samples selection
  seed: 42
