Autotune CISS-VAE hyperparameters with Optuna ...
Usage
autotune_cissvae(
data,
index_col = NULL,
clusters,
save_model_path = NULL,
save_search_space_path = NULL,
n_trials = 20,
study_name = "vae_autotune",
device_preference = "cuda",
show_progress = FALSE,
optuna_dashboard_db = NULL,
load_if_exists = TRUE,
seed = 42,
verbose = FALSE,
num_hidden_layers = c(1, 4),
hidden_dims = c(64, 512),
latent_dim = c(10, 100),
latent_shared = c(TRUE, FALSE),
output_shared = c(TRUE, FALSE),
lr = c(1e-04, 0.001),
decay_factor = c(0.9, 0.999),
beta = 0.01,
num_epochs = 500,
batch_size = 4000,
num_shared_encode = c(0, 1, 3),
num_shared_decode = c(0, 1, 3),
refit_patience = 2,
refit_loops = 100,
epochs_per_loop = 500,
reset_lr_refit = c(TRUE, FALSE)
)