Benchmark multi-model/multi-view models.

Note

This page is a reference documentation. It only explains the function signature, and not how to use it. Please refer to the gallery for the big picture.

mmbench.baseline.vae.train_vae(dataset, datasetdir, outdir, fit_lat_dims=20, n_iter=100, host='http://localhost', port=8085)[source]¶

Train the VAE model.

Parameters

dataset : str

the dataset name: euaims or hbn.

datasetdir : str

the path to the dataset associated data.

outdir : str

the destination folder.

fit_lat_dims : int, default 20

the number of latent dimensions.

n_iter : int, default 100

the number of train iterations.

host : str, default ‘http://localhost’

the host on which visdom is launched.

port : int, default 8085

the port on which the visdom server is launched.