campa.tl.LossEnum

class LossEnum(value)[source]

Loss functions.

Possible values are:

  • LossEnum.MSE: “mean_squared_error” (tf.losses.mean_squared_error())

  • LossEnum.SIGMA_MSE: “sigma_vae_mse”, MSE loss for sigma VAE

  • LossEnum.KL: “kl_divergence”

  • LossEnum.MSE_METRIC: “mean_squared_error_metric” (tf.keras.metrics.MeanSquaredError)