BrandCultivation/report.py

import os
import argparse
from utils import ReportUtils
    
    
def run():
    parser = argparse.ArgumentParser()
    
    parser.add_argument("--city_uuid", type=str, default="00000000000000000000000011445301")
    parser.add_argument("--product_id", type=str, default="350139")
    parser.add_argument("--recall_count", type=int, default=500)
    parser.add_argument("--delivery_count", type=int, default=5000)
    
    # parser.add_argument()
    # parser.add_argument()
    
    args = parser.parse_args()
    
    # 查找该城市的gbdt模型是否存在
    args.gbdtlr_model_path = os.path.join("./models/rank/weights/", args.city_uuid, "gbdtlr_model.pkl")
    if not os.path.exists(args.gbdtlr_model_path):
        print("该城市的模型还未训练，请先启动训练！！！")
        
    # 初始化report生成工具
    report_utils = ReportUtils(args.city_uuid, args.product_id)
    
        
    # 生成报告
    report_utils.generate_all_data(args.recall_count, args.delivery_count)
    
    
if __name__ == "__main__":
    run()