Update loss_functions.rst

bfortuner · Nov 29, 2017 · 54250a4 · 54250a4
1 parent ced172c
commit 54250a4
Showing 1 changed file with 2 additions and 7 deletions.
diff --git a/docs/loss_functions.rst b/docs/loss_functions.rst
@@ -25,13 +25,8 @@ The graph above shows the range of possible loss values given a true observation
 
 .. rubric:: Code
 
-::
-
-  def cross_entropy(true_label, prediction):
-      if true_label == 1:
-          return -log(prediction)
-      else:
-          return -log(1 - prediction)
+.. literalinclude:: ../code/loss_functions.py   
+      :pyobject: CrossEntropy
 
 .. rubric:: Math