Description Usage Arguments Value
View source: R/loss_functions.R
A custom loss function for a VAE learning a standard normal distribution
1 | vae_loss_independent(encoder, kl_weight, rec_dim)
|
encoder |
the encoder model of the VAE, used to obtain z_mean and z_log_var from inputs |
kl_weight |
weight for the KL divergence term |
rec_dim |
the number of nodes in the input/output of the VAE |
returns a function whose parameters match keras loss format
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.