candle.uq_keras_utils.heteroscedastic_loss

candle.uq_keras_utils.heteroscedastic_loss#

candle.uq_keras_utils.heteroscedastic_loss(nout)#

This function computes the heteroscedastic loss for the heteroscedastic model. Both mean and standard deviation predictions are taken into account.

Parameters:

nout (int) – Number of outputs without uq augmentation