From 3d1e8f16ac53aad322f2c681f5f3c25918c58602 Mon Sep 17 00:00:00 2001
From: tslil <tslil@posteo.de>
Date: Tue, 12 Jan 2021 16:48:41 -0500
Subject: Generate the correct format directly

---
 src/train5.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/train5.py')

diff --git a/src/train5.py b/src/train5.py
index 214ad88..adb81a5 100644
--- a/src/train5.py
+++ b/src/train5.py
@@ -69,7 +69,7 @@ def load_data(size):
     training_outcome = training_data.iloc[:, -1:]
 
     val_fn = "validation-"+str(size)+".csv"
-    val_data = pd.read_csv(val_fn).tail(20000)
+    val_data = pd.read_csv(val_fn)  #.tail(20000)
     val_stack_input = np.array(val_data.iloc[:, 2:-1]).reshape(shape, order='F')
     val_flats_input = np.array(val_data.iloc[:, 0:2])
     val_input = [val_stack_input, val_flats_input]
-- 
cgit v1.2.3