LightGBMでverbose_evalとかでUserWarningが出る対策

メッセージ通りに対処すればよい。どこかでちゃんとテンプレ化して置いておきたい。

import lightgbm as lgb

# いろいろ省略

callbacks = [
    lgb.log_evaluation(10),
    lgb.early_stopping(20),
]
gbm = lgb.train(
    params,
    lgb_train,
    valid_sets=lgb_eval,
    num_boost_round=100_000,
    callbacks=callbacks,
)

コメント

タイトルとURLをコピーしました