lightGBMのTypeError: Wrong type(ndarray) for label. It should be list, numpy 1-D array or panda、について

Question

lightGBMを使用していて、
TypeError: Wrong type(ndarray) for label.
It should be list, numpy 1-D array or pandas Series
というerrorが出現してしまいます。

print(X_train.shape) #(1000000, 1)
print(y_train.shape) #(1000000, 1)
print(type(X_train)) #<class 'numpy.ndarray'>
print(type(y_train)) #<class 'numpy.ndarray'>

上記の様なX_train,y_trainを使用して、以下の様に書いたところ上記errorが出てしまいます。

cv = KFold(n_splits=5, shuffle=True, random_state=0)
for fold_id, (train_index,valid_index) in enumerate(cv.split(X_train)):
    X_tr = X_train[train_index] #⭐️
    X_val = X_train[valid_index] #⭐️
    y_tr = y_train[train_index] #⭐️
    y_val = y_train[valid_index]　#⭐️
    
    lgb_train = lgb.Dataset(X_tr,y_tr)
    lgb_val = lgb.Dataset(X_val,y_val)

    model = lgb.train(params,lgb_train,n_estimators,valid_sets=[lgb_train,lgb_val],
                      verbose_eval=50, early_stopping_rounds=200, feval=f1_score_calc )
    
    preds = model.predict(X_val, num_iteration=model.best_iteration)

# TypeError: Wrong type(ndarray) for label.
# It should be list, numpy 1-D array or pandas Series

X_train,y_train共にndarrayですが、shapeが(1000000, 1)であるので2-Dかと考え、
⭐️の箇所を、X_tr = X_train[train_index].ravel()として、全て1-Dに変えてみましたが、すると、

ValueError: Input numpy.ndarray must be 2 dimensional

と今度は2-Dでないといけないかの様なerrorが出てしまいます。
調べても良い解決策が見つからず、分かります方がいらっしゃればご教授頂ければと存じます。
よろしくお願いいたします。

Comments

[kabayan55]

上のエラーと下のエラー、それぞれエラー文を全部貼るとより適切な回答を得やすくなると思います。
エラー文のみですと、どの関数の何行目でエラーを吐いているかを読み取れません。コードだと model.predict の部分なのかなとは思っているのですが、内部の挙動がわからず問題箇所の原因を探るのが難しいです。

Answer 1

It should be list, numpy 1-D array or pandas Series とあって、numpy arrayだとうまくいかないとのことだったので、list あるいは panadas Series で試してみてください。

• リスト

X_train = X_train.tolist()

• Series

import pandas as pd
X_train = pd.Series(X_train)