Commit 8de02106 authored by Alessia Marcolini's avatar Alessia Marcolini
Browse files

Add filename column in HN_BZ dataset

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