From 939b5e83a23f9099df4fc4b65643166867a76019 Mon Sep 17 00:00:00 2001 From: nunzip Date: Sat, 8 Dec 2018 19:15:44 +0000 Subject: Different train split --- evaluate.py | 3 +-- 1 file changed, 1 insertion(+), 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 -- cgit v1.2.3-54-g00ecf