罗俊辉 преди 1 година
родител
ревизия
8ab6ebc166
променени са 1 файла, в които са добавени 4 реда и са изтрити 4 реда
  1. 4 4
      main.py

+ 4 - 4
main.py

@@ -128,7 +128,7 @@ class LightGBM(object):
         评估模型性能
         :return:
         """
-        fw = open("summary_tag.txt", "a+", encoding="utf-8")
+        fw = open("summary_tag_01.txt", "a+", encoding="utf-8")
         # 测试数据
         with open("produce_data/x_data_total_return_predict.json") as f1:
             x_list = json.loads(f1.read())
@@ -181,6 +181,6 @@ class LightGBM(object):
 
 if __name__ == "__main__":
     L = LightGBM()
-    L.train_model()
-    # L.evaluate_model()
-    L.feature_importance()
+    # L.train_model()
+    L.evaluate_model()
+    # L.feature_importance()