Add filename column in HN_BZ dataset

preprocessing/prepare_clinical.py

@@ -15,7 +15,7 @@ from utils import remove_na, remove_constant_cols
 
 # os.chdir('..')
 # %%
-DATASET_NAME = 'HN_val'
+DATASET_NAME = 'HN_BZ'
 PROJECT_DATA_PATH = Path('data') / DATASET_NAME
 RAW_DATA_PATH = PROJECT_DATA_PATH / 'raw'
 PROCESSED_DATA_PATH = PROJECT_DATA_PATH / 'processed'
@@ -384,6 +384,7 @@ elif DATASET_NAME == 'HN_BZ':
     clinical['ROI_name'] = 'lesione'
 
     clinical['ROI_modality'] = 'PT'
+    clinical['filename'] = clinical['patient'].apply(lambda x: x + '.npy')
 
     clinical_no_na.to_csv(
         PROCESSED_DATA_PATH / CLINICAL_DATA_FILENAME_CLEAN, index=False