From e32003145b808d31979474da4117fd902fe1f176 Mon Sep 17 00:00:00 2001 From: JML Alcaraz <68015933+juanlopezcode@users.noreply.github.com> Date: Mon, 12 Aug 2024 18:00:18 +0200 Subject: [PATCH] Update main_ecg.py --- src/main_ecg.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/main_ecg.py b/src/main_ecg.py index 6f3c480..19728a7 100644 --- 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