更新初始化代码

main.py

@@ -42,7 +42,7 @@ float_cols = [
         "out_share_cnt",
         "out_collection_cnt"
     ]
-with open("whole_data/x_data.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)
@@ -60,7 +60,7 @@ with open("whole_data/x_data.json") as f1:
 print("读取X数据成功！")
 
 
-with open("whole_data/y_data.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)