enerzyme.models.loss.NLLLoss

class enerzyme.models.loss.NLLLoss(eps: float = 1e-06, **weights: Dict[str, float])

Bases: WeightedLoss

__init__(eps: float = 1e-06, **weights: Dict[str, float]) → None

loss_fn(output: Dict[str, Tensor], target: Dict[str, Tensor], k: str) → Tensor