规范化训练代码

main.py

@@ -113,7 +113,8 @@ class LightGBM(object):
         num_round = 100
         print("开始训练......")
         bst = lgb.train(params, train_data, num_round, valid_sets=[test_data])
-        bst.save_model(self.model, binary=True)
+        bst.save_model(self.model)
+        print("模型训练完成✅")
 
     def evaluate_model(self):
         """