still trying to get training correct

hlin117 · Jan 1, 2015 · 12b72bc · 12b72bc
1 parent 18c98fb
commit 12b72bc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/demo/lemondetection/main.py b/demo/lemondetection/main.py
@@ -24,7 +24,7 @@ def main():
 
     num_features = len(data[0]) - 1  # Subtract one because of target values
     nn = NeuralNet(num_features, max_epochs=20, learn_rate=.7, scale=0.1, 
-                   hidden_layer=[4], verbose=True, activation="tanh")
+                   hidden_layer=[4], verbose=True, activation=("expit", 2))
     nn.train(features, targets)
 
     print "Done with training. Took {0} seconds to train." \

diff --git a/model/neuralnet.py b/model/neuralnet.py
@@ -5,7 +5,7 @@
 import math
 import numpy as np
 
-one_line = lambda string: string.replace("\n", "").replace("\t", "")
+one_line = lambda string: string.replace("\n", "").replace("  ", "")
 
 class NeuralNet(object):
     """An implementation of a feed forward neural network."""