util

util.py

@@ -85,4 +85,13 @@ def load_data(data_dir : str,
         raise ValueError("Choose mode!")
     
     
-    
+def load_data_and_prepare(data_dir, excel_file, mode, scale, smote):
+    # Load train, validation, and test data
+    train_df,val_df = load_data(data_dir, excel_file, mode, scale, smote)
+    
+    train_df.drop(columns=['patient_id','target'],inplace = True)
+    val_df.drop(columns=['patient_id','target'],inplace = True)
+    
+    train = pd.concat([train_df,val_df],axis=0)
+    
+    return train