Skip to content

Commit 6ac6be1

Browse files
Add reference info to the cross entropy.
1 parent 65743a7 commit 6ac6be1

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

nn.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ def dloss(self, values, expected):
6363
return 2*(values - expected)/values.size
6464

6565
class CrossEntropyLoss(LossFunction):
66+
"""Cross entropy loss function following the pytorch docs."""
6667
def loss(self, values, target_class):
6768
return -values[target_class, 0] + np.log(np.sum(np.exp(values)))
6869

0 commit comments

Comments
 (0)