data
estimators
models
train
versions
ui
return the prediction in the latent space (z_mean for variational models)
Whether toreturn the prediction of z, z_mean, z_log_var in the variational latent space.
latent space