Mercurial > repos > goeckslab > pycaret_predict

diff pycaret_train.py @ 7:f4cb41f458fd 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 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,