summaryrefslogtreecommitdiff
path: root/src/train5.py
diff options
context:
space:
mode:
Diffstat (limited to 'src/train5.py')
-rw-r--r--src/train5.py2
1 files changed, 1 insertions, 1 deletions
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]