kosh/neural-networks-from-scratch
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clean names in loss.py

Browse Source
This commit is contained in:
Koshin S Hegde 2024-05-09 12:31:03 +05:30
parent b6cd28db7c
commit 3fe69b3869
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/loss.py
View File

@ -2,7 +2,7 @@ import numpy as np
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
class Loss(ABC): class LossFunction(ABC):
@staticmethod @staticmethod
@abstractmethod @abstractmethod
def __call__(output: np.ndarray, target: np.ndarray) -> float: def __call__(output: np.ndarray, target: np.ndarray) -> float:
@ -14,7 +14,7 @@ class Loss(ABC):
"""""" """"""
class CrossEntropyLoss(Loss): class CategoricalCrossEntropyLossFunction(LossFunction):
@staticmethod @staticmethod
def __call__(output: np.ndarray, target: np.ndarray) -> float: def __call__(output: np.ndarray, target: np.ndarray) -> float:
return -np.sum(target * np.log10(output), dtype=np.float32) return -np.sum(target * np.log10(output), dtype=np.float32)