qumphy.models.utils.pinballloss module

class qumphy.models.utils.pinballloss.PinballLoss(*args: Any, **kwargs: Any)[source]

Bases: Module

Calculates the quantile loss function.

self.quantiles
Type:

torch.tensor

forward(prediction, target)[source]

Computes the loss for the given prediction.