diff options
| -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 |