indent

1 files changed, 21 insertions, 21 deletions

diff --git a/boston.py b/boston.py
index 6dbd3e7..8d8fe21 100644
--- a/boston.py
+++ b/boston.py
@@ -54,29 +54,29 @@ with tf.device('/cpu:0'):
     # all variables have to be specified here
     sess.run(tf.initialize_all_variables())
 
+    for count in range(0, len(x_train)):
+        trainsteps = 100
+        print("count " + str(count))
+        for i in range(0, trainsteps):  # 100 epochs
+            if False:  # if i % 10 == 9:
+                # TODO mean over test data
+                feed = {
+                    vecInput: x_test[0],
+                    numTarget: [y_test[0]]
+                }
+                result = sess.run([summaries, numMSE], feed_dict=feed)
+                writer.add_summary(result[0], count * trainsteps + i)
 
-for count in range(0, len(x_train)):
-    trainsteps = 100
-    print("count " + str(count))
-    for i in range(0, trainsteps):  # 100 epochs
-        if False:  # if i % 10 == 9:
-            # TODO mean over test data
-            feed = {
-                vecInput: x_test[0],
-                numTarget: [y_test[0]]
-            }
-            result = sess.run([summaries, numMSE], feed_dict=feed)
-            writer.add_summary(result[0], count * trainsteps + i)
+            # TODO run a complete set
+            result = sess.run([summaries, train_step],
+                              feed_dict={
+                                vecInput: x_train[count],
+                                numTarget: [y_train[count]]
+                              })
+            if i % 10 == 9:
+                writer.add_summary(result[0], int(count * trainsteps + (i - 9) / 10))  # TODO this slows down
 
-        # TODO run a complete set
-        result = sess.run([summaries, train_step],
-                          feed_dict={
-                            vecInput: x_train[count],
-                            numTarget: [y_train[count]]
-                          })
-        writer.add_summary(result[0], count * trainsteps + i)  # TODO this slows down
-
-print("finished training")
+    print("finished training")
 
 # yt = [y_test]
 # # debug