分析代码

main.py

@@ -54,7 +54,7 @@ class LightGBM(object):
             "out_share_cnt",
             "out_collection_cnt",
         ]
-        self.split_c = 0.99
+        self.split_c = 0.999
         self.yc = 0.8
         self.model = "lightgbm_tag_train_04.bin"
 
@@ -163,7 +163,7 @@ class LightGBM(object):
         评估模型性能
         :return:
         """
-        fw = open("summary_tag_04.txt", "a+", encoding="utf-8")
+        fw = open("summary_tag_0321.txt", "a+", encoding="utf-8")
         # 测试数据
         with open("produce_data/x_data_total_return_predict.json") as f1:
             x_list = json.loads(f1.read())
@@ -220,6 +220,6 @@ if __name__ == "__main__":
     # study.optimize(L.bays_params, n_trials=100)
     # print('Number of finished trials:', len(study.trials))
     # print('Best trial:', study.best_trial.params)
-    L.train_model()
+    # L.train_model()
     L.evaluate_model()
     L.feature_importance()