罗俊辉 1 rok temu
rodzic
commit
36c2439578
1 zmienionych plików z 2 dodań i 0 usunięć
  1. 2 0
      main.py

+ 2 - 0
main.py

@@ -14,6 +14,8 @@ X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.2, random_
 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(X_train.shape)
+print(X_train)
 print(y_test)
 # 设置模型的参数
 params = {