Skip to content

Commit 78f3576

Browse files
committed
Fix breaking kwarg
1 parent f19d268 commit 78f3576

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

modelbuilders_bench/lgbm_mb.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -118,8 +118,7 @@
118118
t_train, model_lgbm = bench.measure_function_time(lgbm.train, lgbm_params, lgbm_train,
119119
params=params,
120120
num_boost_round=params.n_estimators,
121-
valid_sets=lgbm_train,
122-
verbose_eval=False)
121+
valid_sets=lgbm_train)
123122
train_metric = None
124123
if not X_train.equals(X_test):
125124
y_train_pred = model_lgbm.predict(X_train)

0 commit comments

Comments
 (0)