Update main_ecg.py

AI4HealthUOL · Aug 12, 2024 · e320031 · e320031
1 parent b84fecc
commit e320031
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main_ecg.py b/src/main_ecg.py
@@ -263,7 +263,7 @@ def __init__(self, hparams):
 
             demographics_columns = [i for i in df.columns if 'demographics_' in i]
             biometrics_columns = [i for i in df.columns if 'biometrics_' in i]
-            vitalparameters_columns = [i for i in df.columns if 'vitalparemeters_' in i]
+            vitalparameters_columns = [i for i in df.columns if 'vitals_' in i]
             labvalues_columns = [i for i in df.columns if 'labvalues_' in i]
             all_features = demographics_columns + biometrics_columns + vitalparameters_columns + labvalues_columns
 
@@ -584,7 +584,7 @@ def setup(self, stage):
 
             demographics_columns = [i for i in df_mapped.columns if 'demographics_' in i]
             biometrics_columns = [i for i in df_mapped.columns if 'biometrics_' in i]
-            vitalparameters_columns = [i for i in df_mapped.columns if 'vitalparemeters_' in i]
+            vitalparameters_columns = [i for i in df_mapped.columns if 'vitals_' in i]
             labvalues_columns = [i for i in df_mapped.columns if 'labvalues_' in i]
             all_features = demographics_columns + biometrics_columns + vitalparameters_columns + labvalues_columns