View source: R/loss_functions.R
vae_loss_independent | R Documentation |
A custom loss function for a VAE learning a standard normal distribution
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.