Commit 088cccf2 authored by thecml's avatar thecml
Browse files

fixed small bug in client for make models

parent 775541e3
......@@ -71,12 +71,14 @@ def main():
else:
print("Making dataset with one-hot encoded ats ...")
make_dataset_count.main()
print("\nCompleted generating datasets at:")
print(f"Interim data dictionary: {cfg.INTERIM_DATA_DIR}")
print(f"Processed data dictionary: {cfg.PROCESSED_DATA_DIR}\n")
print(f"Making 4 XGBoost models based on version: {dataset_version} ...\n")
make_xgb_models.main()
make_xgb_models.main(dataset_version)
print(f"Completed making models. Models and SHAP plots can be found at:\n" +
f"{cfg.COMPLETE_XGB_DIR}\n" + f"{cfg.COMPLIANCE_XGB_DIR}\n" +
f"{cfg.FALL_XGB_DIR}\n" + f"{cfg.FALL_TEST_XGB_DIR}" + "\n")
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment