更新初始化代码

main.py

@@ -42,8 +42,9 @@ float_cols = [
         "out_share_cnt",
         "out_collection_cnt"
     ]
-with open("whole_data/x_data_3day_up_level.json") as f1:
+with open("whole_data/x_data_total_return.json") as f1:
     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)
     for key in str_cols:
@@ -59,8 +60,9 @@ with open("whole_data/x_data_3day_up_level.json") as f1:
 print("读取X数据成功！")
 
 
-with open("whole_data/y_data_3day_up_level.json") as f2:
+with open("whole_data/y_data_total_return.json") as f2:
     y_list = json.loads(f2.read())
+    print(len(y_list))
     index_t = int(len(y_list) * 0.7)
     # temp = sorted(y_list)
     # yuzhi = temp[int(len(temp) * 0.8)-1]