Mercurial > repos > goeckslab > pycaret_predict

comparison pycaret_train.py @ 13:f07850192bc2 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 commit 84d5cd0b1fa5c1ff0ad892bc39c95dad1ceb4920
author goeckslab
date Sat, 08 Nov 2025 14:20:33 +0000
parents e2a6fed32d54
children
comparison
equal deleted inserted replaced
12:e674b9e946fb 13:f07850192bc2
118 "--probability_threshold", 118 "--probability_threshold",
119 type=float, 119 type=float,
120 default=None, 120 default=None,
121 help="Probability threshold for classification decision,", 121 help="Probability threshold for classification decision,",
122 ) 122 )
123 parser.add_argument(
124 "--best_model_metric",
125 type=str,
126 default=None,
127 help="Metric used to select the best model (e.g. AUC, Accuracy, R2, RMSE).",
128 )
123 129
124 args = parser.parse_args() 130 args = parser.parse_args()
125 131
126 # Normalize cross-validation flags: --no_cross_validation overrides --cross_validation 132 # Normalize cross-validation flags: --no_cross_validation overrides --cross_validation
127 if args.no_cross_validation: 133 if args.no_cross_validation:
142 "feature_interaction": args.feature_interaction, 148 "feature_interaction": args.feature_interaction,
143 "feature_ratio": args.feature_ratio, 149 "feature_ratio": args.feature_ratio,
144 "fix_imbalance": args.fix_imbalance, 150 "fix_imbalance": args.fix_imbalance,
145 "tune_model": args.tune_model, 151 "tune_model": args.tune_model,
146 "probability_threshold": args.probability_threshold, 152 "probability_threshold": args.probability_threshold,
153 "best_model_metric": args.best_model_metric,
147 } 154 }
148 LOG.info(f"Model kwargs: {model_kwargs}") 155 LOG.info(f"Model kwargs: {model_kwargs}")
149 156
150 # If the XML passed a comma-separated string in a single list element, split it out 157 # If the XML passed a comma-separated string in a single list element, split it out
151 if args.models: 158 if args.models: