Skip to content

Commit 4a28da7

Browse files
committed
Remove deprecated lightGBM args
1 parent 5bea495 commit 4a28da7

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

iris/segmentation/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -377,13 +377,13 @@ def predict_mask(image_id):
377377
tree_learner='data',
378378
learning_rate=0.05,
379379
n_estimators=config['ai_model']['n_estimators'],
380-
silent=True,
381380
n_jobs=10,
382381
)
382+
early_stopping = lgb.early_stopping(4, verbose=False)
383383
gbm.fit(
384384
inputs[train_indices, :], train_labels,
385385
eval_set=[(inputs[val_indices, :], val_labels)],
386-
early_stopping_rounds=4, verbose=0
386+
callbacks=[early_stopping]
387387
)
388388

389389
# predict the mask for the whole image:

0 commit comments

Comments
 (0)