summaryrefslogtreecommitdiff
path: root/src/train5.py
diff options
context:
space:
mode:
authortslil <tslil@posteo.de>2021-01-12 16:48:41 -0500
committertslil <tslil@posteo.de>2021-01-12 16:48:41 -0500
commit82256d3ecb47176721458650fab1f97dabee450e (patch)
treeadb2819f03b34cf119e96d974ee5ae1cec877e06 /src/train5.py
parent5b1a2acd18e9e8eb33d10ba3e219985c13c75f5f (diff)
Generate the correct format directly
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]