OptunaでLightGBMを使おうとしたらtrainがないとエラーになる。

Question

LightGBMをOptunaでチューニングしようとしたのですが、

opt_params = {
    "objective":"regression",
    "metric":"mae"
}

import lightgbm
import optuna.integration.lightgbm as lgbo

opt=lgbo.train(
    opt_params,
    trains,
    valid_sets = valids,
    verbose_eval=False,
    #ラウンド数
    num_boost_round = 5,
    #打ち切り
    early_stopping_rounds = 100
)

このコードを実行したところ、以下のようなエラーが出ます。

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
<ipython-input-47-4c0b4b7f1bdb> in <module>()
      7 import optuna.integration.lightgbm as lgbo
      8 
----> 9 opt=lgbo.train(
     10     opt_params,
     11     trains,

AttributeError: module 'optuna.integration.lightgbm' has no attribute 'train'

また、Dockerで環境を作成しており、imageにFROM gcr.io/kaggle-images/python:v56を利用しています。

原因わかる方いらっしゃいますでしょうか。よろしくお願いします。

Answer 1

ライブラリが古いことが原因でした。新しくすると治りました