In-lined score reporting

1 files changed, 3 insertions, 4 deletions

diff --git a/src/train5.py b/src/train5.py
index 2fc0d6f..3683c1e 100644
--- a/src/train5.py
+++ b/src/train5.py
@@ -100,6 +100,9 @@ def train(size, model, data, iterations=1, epochs=10):
         t_loss, t_accuracy = model.evaluate(training_input, training_outcome,
                                             verbose=False, batch_size=32)
         results += [((v_loss, t_loss), (v_accuracy, t_accuracy))]
+    print("\nScores")
+    for i in range(len(results)):
+        print("Iteration {0}: {1}".format(i+1, results[i]))
     return results
 
 
@@ -163,10 +166,6 @@ def write_weights(model):
 data = load_data(5)
 model = make_model(5, [12, 11, 8])
 results = train(5, model, data, iterations=5, epochs=10)
-print("\nScores")
-for i in range(len(results)):
-    print("Iteration {0}: {1}".format(i+1, results[i]))
-
 write_weights(model)
 
 # results = magic_search(data)