easily make accuracy draw near 50% via reduplicate train

zddhub · Apr 27, 2017 · 9afbdf5 · 9afbdf5
1 parent 6dc70d0
commit 9afbdf5
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/input_data.py b/input_data.py
@@ -44,6 +44,10 @@ def one_hot(self, label):
 		r[self.types.index(label)] = 1
 		return r
 
+	def rewind(self):
+		self.train_i = 0
+		self.test_i = 0
+
 	def train_next_batch(self, size):
 		if self.train_i > len(self.train_indexes):
 			return None, None

diff --git a/trytf.py b/trytf.py
@@ -32,9 +32,12 @@ def main(_):
   tf.global_variables_initializer().run()
 
   # Train
-  for _ in range(180):
-    batch_xs, batch_ys = input_data.train_next_batch(100)
-    sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys})
+  for _ in range(20):
+    for _ in range(90):
+      batch_xs, batch_ys = input_data.train_next_batch(200)
+      sess.run(train_step, feed_dict={x: batch_xs, y_: batch_ys})
+
+    input_data.rewind()
 
   # Test trained model
   correct_prediction = tf.equal(tf.argmax(y, 1), tf.argmax(y_, 1))