yamle.losses.evidential_regression module

Adopted from: aamini/evidential-deep-learning

class yamle.losses.evidential_regression.NIG_NLL(*args, **kwargs)

Bases: Module

Negative log-likelihood loss for Normal Inverse Gamma (NIG) distribution.

forward(y, gamma, v, alpha, beta)

Compute the loss function.

Return type:

Tensor

training: bool

class yamle.losses.evidential_regression.NIG_Reg(*args, **kwargs)

Bases: Module

Regularization loss for Normal Inverse Gamma (NIG) distribution.

forward(y, gamma, v, alpha, beta)

Compute the loss function.

Return type:

Tensor

training: bool

class yamle.losses.evidential_regression.EvidentialRegressionLoss(*args, **kwargs)

Bases: BaseLoss

Evidential regression loss for probabilistic regression.

tasks = ['regression', 'depth_estimation', 'reconstruction']