Python LightGBM:Supported target types are: (‘binary’, ‘multiclass’). Got ‘continuous’ instead.

Published on:
Last updated:

This post is also available in: 日本語 (Japanese)

LightGBMにてCrosss Validation(交差検証)を行っている際に下記のエラーに遭遇しましたので、メモ代わりに書いています。

ValueError: Supported target types are: ('binary', 'multiclass'). Got 'continuous' instead.

今回遭遇したケースでは、層化抽出法(stratified)はクラス分類(classification)で有効で、回帰分析(regression)ではエラーが出るというものでした。
気を付けなければいけない点として、Python API の lightgbm.cv では、デフォルトでstratified=Trueになってますので(Supported target types are...multiclass というエラー内容どおり)、回帰分析(regression)としてデータセットのtarget type(yラベルの事すね)を用意している場合にはstratified=Falseとするとエラーが出なくなるかと思います。

stratified=False

参照サイト:
Python: LightGBM cross validation. How to use lightgbm.cv for regression?

About
Kuniyoshi Takemoto is the founder of Amelt.net LLC, and editor of this blog(www.amelt.net).Learn more and follow me on LinkedIn.