Update util.py

util.py

@@ -36,7 +36,7 @@ def load_data(data_dir : str,
     #    'BILIRUBIN', 'HR_100', 'GGT', 'BUN', 'CREATININE', 'BT_38', 'target']
 
     # Forward (n=13)
-    columns = ['patient_id',  'HR', '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', 'TOTAL_BILIRUBIN',  'target']
     
     # # VISIBLE_STONE_CT (n=1)
     # columns = ['patient_id','VISIBLE_STONE_CT', 'target']