Bläddra i källkod

更新初始化代码

罗俊辉 1 år sedan
förälder
incheckning
5bb4db6a7a
1 ändrade filer med 2 tillägg och 2 borttagningar
  1. 2 2
      main.py

+ 2 - 2
main.py

@@ -43,7 +43,7 @@ float_cols = [
         "out_collection_cnt"
     ]
 with open("whole_data/x_data_3day_up_level.json") as f1:
-    x_list = json.loads(f1.read())[:20000]
+    x_list = json.loads(f1.read())[:30000]
     print(len(x_list))
     index_t = int(len(x_list) * 0.7)
     X_train = pd.DataFrame(x_list[:index_t], columns=my_c)
@@ -61,7 +61,7 @@ print("读取X数据成功!")
 
 
 with open("whole_data/y_data_3day_up_level.json") as f2:
-    y_list = json.loads(f2.read())[:20000]
+    y_list = json.loads(f2.read())[:30000]
     print(len(y_list))
     index_t = int(len(y_list) * 0.7)
     # temp = sorted(y_list)