Update full_preprocessing.py

AI4HealthUOL · Aug 9, 2024 · 12ec246 · 12ec246
1 parent a4dffb3
commit 12ec246
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/full_preprocessing.py b/src/full_preprocessing.py
@@ -166,8 +166,8 @@ def get_diagnosis_ed(subject_id, ecg_time,also_hosp_diag=True):
         unique_intervals = age_bins.cat.categories
         bin_labels = {interval: f'{interval.left}-{interval.right}' for interval in unique_intervals}
         df_full['age_bin'] = age_bins.map(bin_labels)
-        df_full['age_bin'] = df_full['age_bin'].cat.add_categories(['missing']).fillna('missing')
-        df_full['gender'] = df_full['gender'].fillna('missing')
+        df_full['age_bin'] = df_full['age_bin'].cat.add_categories(['missing_age']).fillna('missing_age')
+        df_full['gender'] = df_full['gender'].fillna('missing_gender')
 
         df_full['merged_strat'] = df_full.apply(lambda row: row['label_strat_all2all'] + [row['age_bin'], row['gender']], axis=1)