Upload New File

1_DL_base/Loss.py

+import numpy as np
+
+
+class CrossEntropyLoss:
+
+    def __init__(self):
+        self.prediction_tensor = None
+
+    def forward(self, prediction_tensor, label_tensor):
+        loss = 0
+        a = prediction_tensor * label_tensor
+        for i in range(a.shape[0]):
+            loss = loss + (-np.log(np.sum(a[i]) + np.finfo(float).eps))
+
+        self.prediction_tensor = np.copy(prediction_tensor)
+        return loss
+
+    def backward(self, label_tensor):
+        error_tensor = np.empty(label_tensor.shape)
+        for i in range(label_tensor.shape[0]):
+            error_tensor[i] = - label_tensor[i]/(self.prediction_tensor[i] + np.finfo(float).eps)
+        return error_tensor