Mercurial > repos > goeckslab > pycaret_predict
diff pycaret_train.py @ 7:f4cb41f458fd draft default tip
planemo upload for repository https://github.com/goeckslab/gleam commit b430f8b466655878c3bf63b053655fdbf039ddb0
| author | goeckslab |
|---|---|
| date | Wed, 09 Jul 2025 01:13:01 +0000 |
| parents | ccd798db5abb |
| children |
line wrap: on
line diff
--- a/pycaret_train.py Wed Jul 02 19:00:03 2025 +0000 +++ b/pycaret_train.py Wed Jul 09 01:13:01 2025 +0000 @@ -29,6 +29,9 @@ parser.add_argument("--cross_validation", action="store_true", default=None, help="Perform cross-validation for PyCaret setup") + parser.add_argument("--no_cross_validation", action="store_true", + default=None, + help="Don't perform cross-validation for PyCaret setup") parser.add_argument("--cross_validation_folds", type=int, default=None, help="Number of cross-validation folds \ @@ -62,11 +65,15 @@ args = parser.parse_args() + cross_validation = True + if args.no_cross_validation: + cross_validation = False + model_kwargs = { "train_size": args.train_size, "normalize": args.normalize, "feature_selection": args.feature_selection, - "cross_validation": args.cross_validation, + "cross_validation": cross_validation, "cross_validation_folds": args.cross_validation_folds, "remove_outliers": args.remove_outliers, "remove_multicollinearity": args.remove_multicollinearity,