aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authornunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-08 19:15:44 +0000
committernunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal>2018-12-08 19:15:44 +0000
commit939b5e83a23f9099df4fc4b65643166867a76019 (patch)
tree40195d1b5b23876924033f5d6af0b235d6e2b2ba
parentdca776c1046d5fb493c89c5f9ed6ccbb4c147a5a (diff)
downloadvz215_np1915-939b5e83a23f9099df4fc4b65643166867a76019.tar.gz
vz215_np1915-939b5e83a23f9099df4fc4b65643166867a76019.tar.bz2
vz215_np1915-939b5e83a23f9099df4fc4b65643166867a76019.zip
Different train split
-rwxr-xr-xevaluate.py3
1 files changed, 1 insertions, 2 deletions
diff --git a/evaluate.py b/evaluate.py
index 127b3de..caed403 100755
--- a/evaluate.py
+++ b/evaluate.py
@@ -160,8 +160,7 @@ def main():
cam = cam.reshape((cam.shape[0],1))
labs = labels[train_idx].reshape((labels[train_idx].shape[0],1))
tt = np.hstack((train_idx, cam))
- s_train, s_test, s_train_label, s_test_label = train_test_split(tt, labs, test_size=0.3, random_state=0)
- train, test, train_label, test_label = train_test_split(s_train, s_train_label, test_size=0.3, random_state=0)
+ train, test, train_label, test_label = train_test_split(tt, labs, test_size=0.3, random_state=0)
#to make it smaller we do a double split
del labs
del cam