From 8d6fd7eb69fc4eb5962f4826a163e48accd346fc 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 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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 diff --git a/requirements.txt b/requirements.txt index 4881027..b4c346c 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,6 +1,6 @@ numpy==1.26.4 pandas==2.2.2 -scikit-learn==1.5.1 +scikit-learn==1.2.2 flask==3.0.0 mlflow==2.15.1 joblib==1.4.2