修改gbdt-lr调用接口

gbdt_lr.py

@@ -23,14 +23,8 @@ def train(args):
     print("训练数据整合完成！")
     
     # 进行训练
-    print("开始训练原始模型")
-    trainer(args, os.path.join(args.train_data_dir, "original_train_data.csv"), model_dir, "ori_model.pkl")
-    
-    print("开始训练pos模型")
-    trainer(args, os.path.join(args.train_data_dir, "pos_train_data.csv"), model_dir, "pos_model.pkl")
-    
-    print("开始训练shopping模型")
-    trainer(args, os.path.join(args.train_data_dir, "shopping_train_data.csv"), model_dir, "shopping_model.pkl")
+    print("开始训练gbdt-lr模型")
+    trainer(args, os.path.join(args.train_data_dir, "train_data.csv"), model_dir, "gbdtlr_model.pkl")
 
 def trainer(args, train_data_path, model_dir, model_name):
     trainer = Trainer(train_data_path)