Update code doc for multivariable cost function (#56)

The return value definition was wrong. Also, the arguments didn't match the case of the code doc.
bfortuner · Jan 26, 2019 · 29c804f · 29c804f
1 parent 8febd0e
commit 29c804f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/docs/linear_regression.rst b/docs/linear_regression.rst
@@ -382,10 +382,10 @@ Now we need a cost function to audit how our model is performing. The math is th
 
   def cost_function(features, targets, weights):
       **
-      Features:(200,3)
-      Targets: (200,1)
-      Weights:(3,1)
-      Returns 1D matrix of predictions
+      features:(200,3)
+      targets: (200,1)
+      weights:(3,1)
+      returns average squared error among predictions
       **
       N = len(targets)