Explorar el Código

修复report生成格式

yangzeyu hace 11 meses
padre
commit
468a65eb3a
Se han modificado 1 ficheros con 3 adiciones y 3 borrados
  1. 3 3
      report.py

+ 3 - 3
report.py

@@ -122,7 +122,7 @@ class ReportUtils:
         recommend_data = pd.read_excel(os.path.join(args.report_dir, "商户售卖推荐表.xlsx"))
         report = eval_report_process(eval_order_data, recommend_data)
         
-        report.to_excel(os.path.join(args.report_dir, "效果验证表.xlsx"))
+        report.to_excel(os.path.join(args.report_dir, "效果验证表.xlsx"), index=False)
 
 def generate_all_data(args, report_utils):
     report_utils.generate_feats_ralation_report(args)
@@ -136,8 +136,8 @@ def run():
     parser = argparse.ArgumentParser()
     
     parser.add_argument("--city_uuid", type=str, default="00000000000000000000000011445301")
-    parser.add_argument("--product_id", type=str, default="510149")
-    parser.add_argument("--recall_count", type=int, default=100)
+    parser.add_argument("--product_id", type=str, default="350139")
+    parser.add_argument("--recall_count", type=int, default=1000)
     parser.add_argument("--delivery_count", type=int, default=5000)
     
     # parser.add_argument()