skip to content
barorin&?

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

/ 1 min read

問題

実行環境

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