|
@@ -13,7 +13,8 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_
|
|
|
# 创建LightGBM数据集
|
|
# 创建LightGBM数据集
|
|
|
train_data = lgb.Dataset(X_train, label=y_train)
|
|
train_data = lgb.Dataset(X_train, label=y_train)
|
|
|
test_data = lgb.Dataset(X_test, label=y_test, reference=train_data)
|
|
test_data = lgb.Dataset(X_test, label=y_test, reference=train_data)
|
|
|
-
|
|
|
|
|
|
|
+print(test_data)
|
|
|
|
|
+print(y_test)
|
|
|
# 设置模型的参数
|
|
# 设置模型的参数
|
|
|
params = {
|
|
params = {
|
|
|
'objective': 'binary', # 指定二分类任务
|
|
'objective': 'binary', # 指定二分类任务
|