v

util.py

@@ -30,13 +30,13 @@ def load_data(data_dir : str,
     #Column rename
     df.rename(columns={'ID': 'patient_id', 'REAL_STONE':'target'}, inplace=True)
 
-   # df_all = ['SEX', 'FIRST_SBP', 'FIRST_DBP', 'FIRST_HR', 'FIRST_RR', 'FIRST_BT',
+   # df_all = ['SEX', 'SBP', 'DBP', 'HR', 'RR', 'BT',
     #    'AGE', 'VISIBLE_STONE_CT', 'PANCREATITIS', 'DUCT_DILIATATION_10MM',
     #    'DUCT_DILIATATION_8MM', 'Hb', 'PLT', 'WBC', 'ALP', 'ALT', 'AST', 'CRP',
     #    'BILIRUBIN', 'HR_100', 'GGT', 'BUN', 'CREATININE', 'BT_38', 'target']
 
     # Forward (n=13)
-    columns = ['patient_id',  'FIRST_HR', 'FIRST_BT', 'AGE','DUCT_DILIATATION_10MM', 'Hb','PLT','WBC','ALP', 'ALT', 'AST', 'BILIRUBIN',  'target']
+    columns = ['patient_id',  'HR', 'BT', 'AGE','DUCT_DILIATATION_10MM', 'Hb','PLT','WBC','ALP', 'ALT', 'AST', 'BILIRUBIN',  'target']
     
     # # VISIBLE_STONE_CT (n=1)
     # columns = ['patient_id','VISIBLE_STONE_CT', 'target']