瀏覽代碼

测试提交

罗俊辉 1 年之前
父節點
當前提交
36c2439578
共有 1 個文件被更改,包括 2 次插入0 次删除
  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 = {