Mercurial > repos > goeckslab > pycaret_predict

comparison pycaret_train.py @ 7:f4cb41f458fd draft

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 1aed7d47c5ec
comparison
equal deleted inserted replaced
6:a32ff7201629 7:f4cb41f458fd
27 default=None, 27 default=None,
28 help="Perform feature selection for PyCaret setup") 28 help="Perform feature selection for PyCaret setup")
29 parser.add_argument("--cross_validation", action="store_true", 29 parser.add_argument("--cross_validation", action="store_true",
30 default=None, 30 default=None,
31 help="Perform cross-validation for PyCaret setup") 31 help="Perform cross-validation for PyCaret setup")
32 parser.add_argument("--no_cross_validation", action="store_true",
33 default=None,
34 help="Don't perform cross-validation for PyCaret setup")
32 parser.add_argument("--cross_validation_folds", type=int, 35 parser.add_argument("--cross_validation_folds", type=int,
33 default=None, 36 default=None,
34 help="Number of cross-validation folds \ 37 help="Number of cross-validation folds \
35 for PyCaret setup") 38 for PyCaret setup")
36 parser.add_argument("--remove_outliers", action="store_true", 39 parser.add_argument("--remove_outliers", action="store_true",
60 parser.add_argument("--test_file", type=str, default=None, 63 parser.add_argument("--test_file", type=str, default=None,
61 help="Path to the test data file") 64 help="Path to the test data file")
62 65
63 args = parser.parse_args() 66 args = parser.parse_args()
64 67
68 cross_validation = True
69 if args.no_cross_validation:
70 cross_validation = False
71
65 model_kwargs = { 72 model_kwargs = {
66 "train_size": args.train_size, 73 "train_size": args.train_size,
67 "normalize": args.normalize, 74 "normalize": args.normalize,
68 "feature_selection": args.feature_selection, 75 "feature_selection": args.feature_selection,
69 "cross_validation": args.cross_validation, 76 "cross_validation": cross_validation,
70 "cross_validation_folds": args.cross_validation_folds, 77 "cross_validation_folds": args.cross_validation_folds,
71 "remove_outliers": args.remove_outliers, 78 "remove_outliers": args.remove_outliers,
72 "remove_multicollinearity": args.remove_multicollinearity, 79 "remove_multicollinearity": args.remove_multicollinearity,
73 "polynomial_features": args.polynomial_features, 80 "polynomial_features": args.polynomial_features,
74 "feature_interaction": args.feature_interaction, 81 "feature_interaction": args.feature_interaction,