candle.uq_keras_utils.r2_heteroscedastic_metric

candle.uq_keras_utils.r2_heteroscedastic_metric#

candle.uq_keras_utils.r2_heteroscedastic_metric(nout)#

This function computes the r2 for the heteroscedastic model. The r2 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