Browse Source

分析代码

罗俊辉 1 year ago
parent
commit
6cd1f80691
1 changed files with 2 additions and 2 deletions
  1. 2 2
      main.py

+ 2 - 2
main.py

@@ -167,11 +167,11 @@ class LightGBM(object):
         """
         fw = open("summary_tag_03{}.txt".format(self.dt), "a+", encoding="utf-8")
         # 测试数据
-        with open("produce_data/x_data_total_return_predict.json") as f1:
+        with open("data/produce_data/x_data_total_return_predict_{}.json".format(self.dt)) as f1:
             x_list = json.loads(f1.read())
 
         # 测试 label
-        with open("produce_data/y_data_total_return_predict.json") as f2:
+        with open("data/produce_data/y_data_total_return_predict_{}.json".format(self.dt)) as f2:
             Y_test = json.loads(f2.read())
 
         Y_test = [0 if i <= 27 else 1 for i in Y_test]