Commit ced520e8 authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Do not read clinical file as str

parent 456b7009
...@@ -88,7 +88,7 @@ class NumpyCSVDataset(Dataset): ...@@ -88,7 +88,7 @@ class NumpyCSVDataset(Dataset):
np.random.seed(self.seed) np.random.seed(self.seed)
clinical = pd.read_csv(clinical_file, dtype=str).sort_values(by=['patient']) clinical = pd.read_csv(clinical_file).sort_values(by=['patient'])
available_files = [f for f in os.listdir(data_dir) if f.endswith('.npy')] available_files = [f for f in os.listdir(data_dir) if f.endswith('.npy')]
......
Markdown is supported
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