Added logreg_predictHEADmaster

1 files changed, 3 insertions, 3 deletions

diff --git a/src/logreg_train.py b/src/logreg_train.py
index e02e101..6254494 100755
--- a/src/logreg_train.py
+++ b/src/logreg_train.py
@@ -35,7 +35,7 @@ def train(ys, xs):
         ys_ally = ys.copy()
         ys_ally[ys == trained] = 0 # opposite?
         ys_ally[ys != trained] = 1
-        thetas.append((trained, gradient_descent(ys_ally, xs, 1, 2)))
+        thetas.append((trained, gradient_descent(ys_ally, xs, 1, 100)))
     return thetas
 
 if __name__ == '__main__':
@@ -51,5 +51,5 @@ if __name__ == '__main__':
     thetas = train(Y, X)
 
     with open("weights", "w") as f:
-        for name, t in thetas:
-            f.write("{}: {}\n".format(name, ','.join([str(x) for x in t])))
+        for label, t in thetas:
+            f.write("{}: {}\n".format(label, ','.join([str(x) for x in t])))