소스 검색

更新初始化代码

罗俊辉 1 년 전
부모
커밋
87503b20ce
1개의 변경된 파일2개의 추가작업 그리고 2개의 파일을 삭제
  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())[:390000]
+    x_list = json.loads(f1.read())
     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())[:390000]
+    y_list = json.loads(f2.read())
     print(len(y_list))
     index_t = int(len(y_list) * 0.7)
     # temp = sorted(y_list)