Commit 4b5d8d22 authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Change parameter name from 'label_file' to 'clinical_file'

parent 798aa7bd
......@@ -68,7 +68,7 @@ class NumpyCSVDataset(Dataset):
def __init__(
self,
data_dir,
label_file,
clinical_file,
label_col,
size,
transforms=augment_3D,
......@@ -88,7 +88,7 @@ class NumpyCSVDataset(Dataset):
np.random.seed(self.seed)
clinical = pd.read_csv(label_file, dtype=str).sort_values(by=['patient'])
clinical = pd.read_csv(clinical_file, dtype=str).sort_values(by=['patient'])
available_files = [f for f in os.listdir(data_dir) if f.endswith('.npy')]
......
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