candle.uq_keras_utils.mse_heteroscedastic_metric

candle.uq_keras_utils.mse_heteroscedastic_metric(nout)

This function computes the mean squared error (mse) for the heteroscedastic model. The mse is computed over the prediction of the mean and the standard deviation prediction is not taken into account.

Parameters:

nout (int) – Number of outputs without uq augmentation