Fix: wrong place to configure max_features

ThoughtWorksInc · Aug 20, 2024 · 8d6fd7e · 8d6fd7e
1 parent be35a36
commit 8d6fd7e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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