Mercurial > repos > goeckslab > image_learner

diff image_learner_cli.py @ 18:bbf30253c99f draft default tip

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
planemo upload for repository https://github.com/goeckslab/gleam.git commit c155acd3616dfac920b17653179d7bc38ba48e39
author goeckslab
date Sun, 14 Dec 2025 03:27:12 +0000
parents d17e3a1b8659
children
line wrap: on
line diff
--- a/image_learner_cli.py	Wed Dec 10 00:24:13 2025 +0000
+++ b/image_learner_cli.py	Sun Dec 14 03:27:12 2025 +0000
@@ -145,26 +145,11 @@
     parser.add_argument(
         "--validation-metric",
         type=str,
-        default="roc_auc",
-        choices=[
-            "accuracy",
-            "loss",
-            "roc_auc",
-            "balanced_accuracy",
-            "precision",
-            "recall",
-            "f1",
-            "specificity",
-            "log_loss",
-            "pearson_r",
-            "mae",
-            "mse",
-            "rmse",
-            "mape",
-            "r2",
-            "explained_variance",
-        ],
-        help="Metric Ludwig uses to select the best model during training/validation.",
+        default=None,
+        help=(
+            "Metric Ludwig uses to select the best model during training/validation. "
+            "Leave unset to let the tool pick a default for the inferred task."
+        ),
     )
     parser.add_argument(
         "--target-column",