diff --git a/src/loss.py b/src/loss.py index f73ad86..4ccd105 100644 --- a/src/loss.py +++ b/src/loss.py @@ -2,7 +2,7 @@ import numpy as np from abc import ABC, abstractmethod -class Loss(ABC): +class LossFunction(ABC): @staticmethod @abstractmethod def __call__(output: np.ndarray, target: np.ndarray) -> float: @@ -14,7 +14,7 @@ class Loss(ABC): """""" -class CrossEntropyLoss(Loss): +class CategoricalCrossEntropyLossFunction(LossFunction): @staticmethod def __call__(output: np.ndarray, target: np.ndarray) -> float: return -np.sum(target * np.log10(output), dtype=np.float32)