Mercurial > repos > goeckslab > image_learner

comparison 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
comparison
equal deleted inserted replaced
17:db9be962dc13 18:bbf30253c99f
143 ), 143 ),
144 ) 144 )
145 parser.add_argument( 145 parser.add_argument(
146 "--validation-metric", 146 "--validation-metric",
147 type=str, 147 type=str,
148 default="roc_auc", 148 default=None,
149 choices=[ 149 help=(
150 "accuracy", 150 "Metric Ludwig uses to select the best model during training/validation. "
151 "loss", 151 "Leave unset to let the tool pick a default for the inferred task."
152 "roc_auc", 152 ),
153 "balanced_accuracy",
154 "precision",
155 "recall",
156 "f1",
157 "specificity",
158 "log_loss",
159 "pearson_r",
160 "mae",
161 "mse",
162 "rmse",
163 "mape",
164 "r2",
165 "explained_variance",
166 ],
167 help="Metric Ludwig uses to select the best model during training/validation.",
168 ) 153 )
169 parser.add_argument( 154 parser.add_argument(
170 "--target-column", 155 "--target-column",
171 type=str, 156 type=str,
172 default=None, 157 default=None,