Remove unused variable

3ad9c75b · Alessia Marcolini · e10d2010 · 3ad9c75b
Commit 3ad9c75b authored Nov 07, 2019 by Alessia Marcolini
--- a/split.py
+++ b/split.py
@@ -2,8 +2,7 @@ import numpy as np
 from sklearn.model_selection import train_test_split

 def train_test_indexes_patient_wise(dataset, test_size=0.2, stratify=True):
-     
-    files = dataset.get_files()
+    
    patients = dataset.get_patients()
    unique_patients = np.unique(patients)