勾配ブースティング決定木の学習時のy_trainの次元の扱い

Question

fitの引数の次元をどう修正すれば教えていただきたいです

勾配ブースティング決定木の学習をしようと思っているのですが、y_trainの部分にエラーが発生しているらしいので、この解決方法を教えていただきたいです。

この処理後にエラーが発生します。

gbr = GradientBoostingRegressor(random_state=0)
gbr.fit(X_train,y_train)

y_trainの中身は、先頭５行で

print(y_train.head(5))

    kpl
0  34.6
1  24.0
2  30.3
3  39.2
4  23.5

となっています。

発生している問題・エラーメッセージ

DataConversionWarning: A column-vector y was passed when a 1d array was expected. Please change the shape of y to (n_samples, ), for example using ravel().
  y = column_or_1d(y, warn=True)

試したこと

言われた通りDataFrame型のy_trainを１次元にするべきと思い、

y_train.ravel() #DataFrame型なので、使用不可でした
y_train = y_train.values.tolist()  #ndarrayを経由してリスト化

をしましたが、同じようなエラーでした。

よろしくお願いいたします。

Answer 1

y_train.valuesのshapeを見てみて、そのままで平気そうなら（普通は平気だと思います）これを直接、もし必要ならravelというのが無難かと。

Comments

[OkomekO]

無事動かすことができました！これから　のパラメータ調節が楽しみですありがとうございました！！