I am trying to replace nLL with the Heteroscedastic uncertainty model using the formula given here which expects the model to output the the predicted mean and variance.
I know that the output returns a tuple of predicted mean and KL but is there a way to get the predicted variance?
I am trying to replace nLL with the Heteroscedastic uncertainty model using the formula given here which expects the model to output the the predicted mean and variance. I know that the output returns a tuple of predicted mean and KL but is there a way to get the predicted variance?