enerzyme.models.loss.WeightedLoss#

class enerzyme.models.loss.WeightedLoss(**weights: Dict[str, float])[source]#

Bases: ABC

__init__(**weights: Dict[str, float]) → None[source]#

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