【LightGBM】予測値が2種類出てしまう

Question

kaggleでデータサイエンスの勉強をしています。以下がnotebookです
https://www.kaggle.com/tsukasatsukano/notebook78e2776b23

　LightBGMを用いて学習と予測を行いました。予測値であるtargetの値は１種類だけです。しかし、自分が予測値として出力したy_predは２種類の値を出力してしまいました。何が原因なのか教えていただきたいです。よろしくお願いいたします
ちなみにセル１６は勉強のために書いたコードで特に意味はないです

Comments

[toast-uz]

リンク先のnotebookが空になってませんか？

[tsukano]

すみませんでした。コードを編集して問題は解決しました。次回以降気を付けます

Answer 1

LGBMClassifier()をLGBMRegressor()にしてコードの修正を行い解決しました