fix

ML/model_train.py

@@ -115,7 +115,7 @@ def create_dataset():
     home_loan_val_final.to_csv('home_loan_val.csv', index=False)
 
 def main() -> None:
-    create_dataset()
+    # create_dataset()
     parser = argparse.ArgumentParser(description='A test program.')
     parser.add_argument("--epochs", help="Prints the supplied argument.", default='10')
     args = parser.parse_args()