diff --git a/bitten/bitten.py b/bitten/bitten.py index b05bc3d..bf64cc9 100644 --- a/bitten/bitten.py +++ b/bitten/bitten.py @@ -262,7 +262,7 @@ class Accuracy(AbstractError): class Crossentropy(AbstractError): - def crossentropy(self, predictions, eps=1e-6): + def _compute(self, predictions, eps=1e-6): p, t = np.clip(predictions, eps, 1 - eps), self.outputs f = np.sum(-t * np.log(p) - (1 - t) * np.log(1 - p), axis=-1) return np.mean(f)