optunaでPruningする時に早すぎる枝刈りをされないようにする方法

課題

optunaでPruningを利用するときにデフォルトの設定だと、0step目でも遠慮なく枝切りされてしまう。
最初の数ステップはあまり差が起きない場合にこれは都合が悪い。

解決策

Prunerにn_warmup_stepsを設定する。

例

study = optuna.create_study()  # Create a new study.

本家のexampleでこう書いてある部分を。

study = optuna.create_study(pruner=ptuna.pruners.MedianPruner(n_warmup_steps=10))  # Create a new study.

こう設定します。(この場合は10stepまでは必ず実行するというものになります)