diff options
| author | nunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal> | 2018-12-08 19:15:44 +0000 |
|---|---|---|
| committer | nunzip <np_scarh@e4-pattern-vm.europe-west4-a.c.electric-orbit-223819.internal> | 2018-12-08 19:15:44 +0000 |
| commit | 939b5e83a23f9099df4fc4b65643166867a76019 (patch) | |
| tree | 40195d1b5b23876924033f5d6af0b235d6e2b2ba /evaluate.py | |
| parent | dca776c1046d5fb493c89c5f9ed6ccbb4c147a5a (diff) | |
| download | vz215_np1915-939b5e83a23f9099df4fc4b65643166867a76019.tar.gz vz215_np1915-939b5e83a23f9099df4fc4b65643166867a76019.tar.bz2 vz215_np1915-939b5e83a23f9099df4fc4b65643166867a76019.zip | |
Different train split
Diffstat (limited to 'evaluate.py')
| -rwxr-xr-x | evaluate.py | 3 |
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 |