From 145fa5fac8e5ddcee07128d9cd2296fea804afe4 Mon Sep 17 00:00:00 2001 From: Luiz Augusto Machado Date: Tue, 20 Aug 2024 17:05:17 -0300 Subject: [PATCH] Fix: wrong place to configure max_features --- cd4ml/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/cd4ml/train.py b/cd4ml/train.py index 073db07..07db03c 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, max_features=5) + algorithm_params, seed=seed) return trained_model