Skip to content

Commit 145fa5f

Browse files
committed
Fix: wrong place to configure max_features
1 parent be35a36 commit 145fa5f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cd4ml/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -26,6 +26,6 @@ def get_trained_model(algorithm_name,
2626
logger.info('n_rows: %s, n_cols: %s' % (n_rows, n_cols))
2727

2828
trained_model = train_model(encoded_train_data, target_data, algorithm_name,
29-
algorithm_params, seed=seed, max_features=5)
29+
algorithm_params, seed=seed)
3030

3131
return trained_model

0 commit comments

Comments
 (0)