Commit 3ad9c75b authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Remove unused variable

parent e10d2010
......@@ -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)
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment