Mercurial > repos > goeckslab > image_learner

comparison image_learner_cli.py @ 15:d17e3a1b8659 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 bc50fef8acb44aca15d0a1746e6c0c967da5bb17
author goeckslab
date Fri, 28 Nov 2025 15:45:49 +0000
parents bcfa2e234a80
children
comparison
equal deleted inserted replaced
14:94cd9ac4a9b1 15:d17e3a1b8659
140 help=( 140 help=(
141 "Decision threshold for binary classification (0.0–1.0)." 141 "Decision threshold for binary classification (0.0–1.0)."
142 "Overrides default 0.5." 142 "Overrides default 0.5."
143 ), 143 ),
144 ) 144 )
145 parser.add_argument(
146 "--validation-metric",
147 type=str,
148 default="roc_auc",
149 choices=[
150 "accuracy",
151 "loss",
152 "roc_auc",
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 )
169 parser.add_argument(
170 "--target-column",
171 type=str,
172 default=None,
173 help="Name of the target/label column in the metadata file (defaults to 'label').",
174 )
175 parser.add_argument(
176 "--image-column",
177 type=str,
178 default=None,
179 help="Name of the image column in the metadata file (defaults to 'image_path').",
180 )
145 181
146 args = parser.parse_args() 182 args = parser.parse_args()
147 183
148 argument_checker(args, parser) 184 argument_checker(args, parser)
149 185