分析代码

main.py

@@ -165,7 +165,7 @@ class LightGBM(object):
         评估模型性能
         :return:
         """
-        fw = open("summary_tag_0319.txt", "a+", encoding="utf-8")
+        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:
             x_list = json.loads(f1.read())
@@ -174,7 +174,7 @@ class LightGBM(object):
         with open("produce_data/y_data_total_return_predict.json") as f2:
             Y_test = json.loads(f2.read())
 
-        Y_test = [0 if i <= 26 else 1 for i in Y_test]
+        Y_test = [0 if i <= 27 else 1 for i in Y_test]
         X_test = pd.DataFrame(x_list, columns=self.my_c)
         for key in self.str_columns:
             X_test[key] = self.label_encoder.fit_transform(X_test[key])