diff pycaret_train.py @ 9:e7dd78077b72 draft

planemo upload for repository https://github.com/goeckslab/gleam commit 84d5cd0b1fa5c1ff0ad892bc39c95dad1ceb4920

--- a/pycaret_train.py	Mon Sep 08 22:38:55 2025 +0000
+++ b/pycaret_train.py	Sat Nov 08 14:20:19 2025 +0000
@@ -120,6 +120,12 @@
         default=None,
         help="Probability threshold for classification decision,",
     )
+    parser.add_argument(
+        "--best_model_metric",
+        type=str,
+        default=None,
+        help="Metric used to select the best model (e.g. AUC, Accuracy, R2, RMSE).",
+    )
 
     args = parser.parse_args()
 
@@ -144,6 +150,7 @@
         "fix_imbalance": args.fix_imbalance,
         "tune_model": args.tune_model,
         "probability_threshold": args.probability_threshold,
+        "best_model_metric": args.best_model_metric,
     }
     LOG.info(f"Model kwargs: {model_kwargs}")