From be35a36dfb70c4e43c8dce51622169dcf207c9e1 Mon Sep 17 00:00:00 2001 From: Luiz Augusto Machado Date: Tue, 20 Aug 2024 16:15:02 -0300 Subject: [PATCH] Teste: wSet a value to max_feature --- cd4ml/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cd4ml/train.py b/cd4ml/train.py index 07db03c..073db07 100644 --- a/cd4ml/train.py +++ b/cd4ml/train.py @@ -26,6 +26,6 @@ def get_trained_model(algorithm_name, logger.info('n_rows: %s, n_cols: %s' % (n_rows, n_cols)) trained_model = train_model(encoded_train_data, target_data, algorithm_name, - algorithm_params, seed=seed) + algorithm_params, seed=seed, max_features=5) return trained_model