小编dee*_*eep的帖子

LightGBM错误:ValueError:为了提前停止,至少需要一个数据集和评估指标进行评估

我正在尝试使用 gridsearch 训练 LightGBM,当我尝试训练模型时出现以下错误。

ValueError: For early stopping, at least one dataset and eval metric is required for evaluation
Run Code Online (Sandbox Code Playgroud)

我提供了验证数据集和评估指标。不知道为什么我仍然遇到这个问题。这是我的代码。

train_data  = rtotal[rtotal['train_Y'] == 1]
test_data   = rtotal[rtotal['train_Y'] == 0]

trainData, validData = train_test_split(train_data, test_size=0.007, random_state = 123)

#train data prep
X_train = trainData.iloc[:,2:71]
y_train = trainData.loc[:,['a_class']]

#validation data prep
X_valid = validData.iloc[:,2:71]
y_valid = validData.loc[:,['a_class']]

#X_test
X_test = test_data.iloc[:,2:71]

import lightgbm as lgb
from sklearn.model_selection import GridSearchCV

gridParams = {
    'learning_rate': [0.005],
    'n_estimators': [40],
    'num_leaves': [16,32, 64],
    'objective' : …
Run Code Online (Sandbox Code Playgroud)

python machine-learning python-3.x scikit-learn lightgbm

5
推荐指数
1
解决办法
2万
查看次数