python - 在 python 中使用 LightGBM 包时正确分配标签类型

Question

我正在尝试在 Python 中使用 LightGBM 包并遇到此错误:

“TypeError:标签的错误类型(ndarray)，应该是列表或 numpy 数组”。

我的目标(标签)被创建为:y_train.values 并且是一个具有如下特征的数组:

Type: int64,
Size: (1000,1)
Value: array([[0],
              [0],
...)

当我追溯这个错误时，我发现了这段代码@ Basic.py code of lightgbm package :

函数 list_to_1d_numpy 抛出此错误。

虽然这个函数应该抛出错误，但我找不到任何理由。然而它正在调用一个检查条件的函数 is_numpy_1d_array

len(data.shape) == 1，但是当我执行 len(y_train.shape) 时，它显示为 2。

有什么办法可以解决吗？

Answer 1

好的，我的想法是正确的。 label(y_train) 需要是一维数组。我通过使用将其更改为一维:

y=y_train.ravel()

成功了!

然而，在创建目标本身时，我们本可以拥有

y_train=dataframe['target'].values
I had it like: dataframe[['target']].values,

创建二维数组