罗俊辉 1 год назад
Родитель
Сommit
ec963b34b9
2 измененных файлов с 7 добавлено и 1 удалено
  1. 5 0
      functions/generate_data.py
  2. 2 1
      main.py

+ 5 - 0
functions/generate_data.py

@@ -0,0 +1,5 @@
+"""
+Created on Mon Mar 18, 2024
+@author: luojunhui
+"""
+

+ 2 - 1
main.py

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