Teste: wSet a value to max_feature

ThoughtWorksInc · Aug 20, 2024 · be35a36 · be35a36
1 parent b2f77b1
commit be35a36
Showing 1 changed file with 1 addition and 1 deletion.
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)
+                                algorithm_params, seed=seed, max_features=5)
 
     return trained_model