diff tests/test_logistic_regression.py @ 23:f34110e28a0a

[logistic regression] we have a working cost function
author Jeff Hammel <k0scist@gmail.com>
date Mon, 04 Sep 2017 09:14:25 -0700
parents 3713c6733990
children 77f68c241b37
line wrap: on
line diff
--- a/tests/test_logistic_regression.py	Mon Sep 04 08:59:52 2017 -0700
+++ b/tests/test_logistic_regression.py	Mon Sep 04 09:14:25 2017 -0700
@@ -21,7 +21,7 @@
 
         expected_cost = 6.000064773192205
         cost = logistic_regression.cost_function(w, b, X, Y)
-        print cost
+        assert abs(cost - expected_cost) < 1e-6
 
 if __name__ == '__main__':
     unittest.main()