Pythonと!

プログラミングド素人の勉強日記です。

AttributeError: module 'lightgbm' has no attribute 'Dataset'になった話

環境:
Windows10, Anaconda, Jupyter
python3.7.6
lightgbm 2.3.1

import lightgbm as lgb

lgb_train = lgb.Dataset(X_train, y_train)
lgb_eval = lgb.Dataset(X_test, y_test, reference=lgb_train)

としたところ、
「AttributeError: module 'lightgbm' has no attribute 'Dataset'」
というエラーが出てしまいました。

解決策?として、以下のimportを明示的に追加したらうまく動作しました。

from lightgbm import Dataset