BrandCultivation/models/rank/data/config.py

class CustConfig:
    FEATURE_COLUMNS = [
        "BB_RETAIL_CUSTOMER_CODE",                     # 零售户代码
        "BB_RTL_CUST_POSITION_TYPE_NAME",              # 零售户商圈类型名称
        "BB_RTL_CUST_MARKET_TYPE_NAME",                # 零售户市场类型名称
        # "BB_RTL_CUST_BUSINESS_TYPE_NAME",              # 零售户业态名称
        "BB_RTL_CUST_SUB_BUSI_PLACE_NAME",             # 零售户业态细分名称
        "BB_RTL_CUST_GRADE_NAME",                      # 零售户分档名称
        # "BB_RTL_CUST_TERMINAL_LEVEL_NAME",             # 零售户终端层级名称
        # "BB_RTL_CUST_TERMINALEVEL_NAME",               # 零售户终端层级细分名称
        # "MD04_MG_RTL_CUST_CREDITCLASS_NAME",           # 零售户信用等级名称
        # "MD04_MG_SAMPLE_CUST_FLAG",                    # 样本户标识
        # "MD07_RTL_CUST_IS_SALE_LARGE_FLAG",            # 零售户大户标识
        # "BB_CUSTOMER_MANAGER_SCOPE_NAME",              # 零售户经营范围名称
        # "BB_RTL_CUST_OPERATE_METHOD_NAME",             # 零售户经营方式名称
        # "BB_RTL_CUST_CGT_OPERATE_SCOPE_NAME",          # 零售户卷烟经营规模名称
        "BB_RTL_CUST_CHAIN_FLAG",                      # 零售户连锁标识
        "MD04_DIR_SAL_STORE_FLAG",                     # 直营店标识
        "STORE_AREA",                                  # 店铺经营面积
        "OPERATOR_AGE",                                # 经营者年龄
        "OPERATOR_EDU_LEVEL",                          # 零售客户经营者文化程
        # "AVERAGE_CONSUMER_FLOW",                       # 月均消费人流
        # "NEW_PRODUCT_MEMBERS_QTY",                     # 新品消费会员数量
    ]
    # 数据清洗规则
    CLEANING_RULES = {
        "BB_RTL_CUST_POSITION_TYPE_NAME":           {"method": "fillna", "opt": "fill", "value": "其他", "type": "str"},
        "BB_RTL_CUST_MARKET_TYPE_NAME":             {"method": "fillna", "opt": "fill", "value": "城网", "type": "str"},
        "BB_RTL_CUST_SUB_BUSI_PLACE_NAME":          {"method": "fillna", "opt": "fill", "value": "其他", "type": "str"},
        "BB_RTL_CUST_GRADE_NAME":                   {"method": "fillna", "opt": "fill", "value": "十五档", "type": "str"},
        # "BB_RTL_CUST_TERMINALEVEL_NAME":          {"method": "fillna", "opt": "replace", "value": "BB_RTL_CUST_TERMINAL_LEVEL_NAME", "type": "str"},
        # "MD04_MG_RTL_CUST_CREDITCLASS_NAME":        {"method": "fillna", "opt": "fill", "value": "未评价", "type": "str"},
        # "MD04_MG_SAMPLE_CUST_FLAG":                 {"method": "fillna", "value": "N", "opt": "fill"},
        # "MD07_RTL_CUST_IS_SALE_LARGE_FLAG":         {"method": "fillna", "value": "N", "opt": "fill"},
        # "BB_RTL_CUST_CGT_OPERATE_SCOPE_NAME":       {"method": "fillna", "value": "中", "opt": "fill"},
        "BB_RTL_CUST_CHAIN_FLAG":                   {"method": "fillna", "opt": "fill", "value": "否", "type": "str"},
        "MD04_DIR_SAL_STORE_FLAG":                  {"method": "fillna", "opt": "fill", "value": "否", "type": "str"},
        "STORE_AREA":                               {"method": "fillna", "opt": "mean", "type": "num"},
        "OPERATOR_AGE":                             {"method": "fillna", "opt": "mean", "type": "num"},
        "OPERATOR_EDU_LEVEL":                       {"method": "fillna", "opt": "fill", "value": "01", "type": "str"},
    }
    # one-hot编码
    ONEHOT = [
        "BB_RTL_CUST_POSITION_TYPE_NAME",
        "BB_RTL_CUST_MARKET_TYPE_NAME",
        "BB_RTL_CUST_SUB_BUSI_PLACE_NAME",
        "BB_RTL_CUST_GRADE_NAME",
        "BB_RTL_CUST_CHAIN_FLAG",
        "MD04_DIR_SAL_STORE_FLAG",
        "OPERATOR_EDU_LEVEL",
    ]
    
    ONEHOT_CAT = {
        "BB_RTL_CUST_POSITION_TYPE_NAME":           ["居民区", "商业娱乐区", "交通枢纽区", "旅游景区", "工业区", "集贸区", "院校学区", "办公区", "其他"],
        "BB_RTL_CUST_MARKET_TYPE_NAME":             ["城网", "农网"],
        "BB_RTL_CUST_SUB_BUSI_PLACE_NAME":          ["便利店", "超市", "烟草专业店", "娱乐服务类", "其他"],
        "BB_RTL_CUST_GRADE_NAME":                   ['一档', '二档', '三档', '四档', '五档', '六档', '七档', '八档', '九档', '十档', '十一档', '十二档', 
                                                    '十三档', '十四档', '十五档', '十六档', '十七档', '十八档', '十九档', '二十档', '二十一档', '二十二档', 
                                                    '二十三档', '二十四档', '二十五档', '二十六档', '二十七档', '二十八档', '二十九档', '三十档'],
        "BB_RTL_CUST_CHAIN_FLAG":                   ["是", "否"],
        "MD04_DIR_SAL_STORE_FLAG":                  ["是", "否"],
        "OPERATOR_EDU_LEVEL":                       [1, 2, 3, 4, 5, 6, 7]
    }
    
    
    
class ProductConfig:
    FEATURE_COLUMNS = [
        "product_code",                                # 商品编码
        # "factory_name",                                # 工业公司名称
        # "brand_code",                                  # 品牌编码
        "adjust_price",                                # 含税调拨价
        "notwithtax_adjust_price",                     # 不含税调拨价
        "whole_sale_price",                            # 统一批发价
        "direct_retail_price",                         # 建议零售价
        "allot_price",                                 # 调拨价
        "direct_whole_price",                          # 批发指导价
        "retail_price",                                # 零售价
        "price_type_name",                             # 卷烟价类名称
        "gear_type_name",                              # 卷烟档位名称
        "category_type_name",                          # 卷烟品类名称
        "is_key_brand",                                # 是否重点品牌
        "is_high_level",                               # 是否高端烟
        "is_upscale_level",                            # 是否高端烟不含高价
        "is_high_price",                               # 是否高价烟
        "is_low_price",                                # 是否低价烟
        "is_low_tar",                                  # 是否低焦油烟
        "is_encourage",                                # 是否全国鼓励品牌
        "is_abnormity",                                # 是否异形包装
        "is_intake",                                   # 是否进口烟
        "is_short",                                    # 是否紧俏品牌
        "is_medium",                                   # 是否中支烟
        "is_shortbranch",                              # 是否短支烟
        "is_ordinary_price_type",                      # 是否普一类烟
        "source_type",                                 # 来源类型
        "tar_qty",                                     # 焦油含量
        "product_style_code_name",                     # 包装类型名称
        "chinese_mix",                                 # 中式混合
        "sub_price_type_name",                         # 细分卷烟价类名称
    ]
    
    CLEANING_RULES = {
        "adjust_price":                                {"method": "fillna", "opt": "mean", "type": "num"},
        "notwithtax_adjust_price":                     {"method": "fillna", "opt": "mean", "type": "num"},
        "whole_sale_price":                            {"method": "fillna", "opt": "mean", "type": "num"},
        "direct_retail_price":                         {"method": "fillna", "opt": "mean", "type": "num"},
        "allot_price":                                 {"method": "fillna", "opt": "fill", "type": "num", "value": 0.0},
        "direct_whole_price":                          {"method": "fillna", "opt": "mean", "type": "num"},
        "retail_price":                                {"method": "fillna", "opt": "mean", "type": "num"},
        "price_type_name":                             {"method": "fillna", "opt": "fill", "type": "str", "value": "一类烟"},
        "gear_type_name":                              {"method": "fillna", "opt": "fill", "type": "str", "value": "其他"},
        "category_type_name":                          {"method": "fillna", "opt": "fill", "type": "str", "value": "其他"},
        "is_key_brand":                                {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_high_level":                               {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_upscale_level":                            {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_high_price":                               {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_low_price":                                {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_low_tar":                                  {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_encourage":                                {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_abnormity":                                {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_intake":                                   {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_short":                                    {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_medium":                                   {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_shortbranch":                              {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "is_ordinary_price_type":                      {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "source_type":                                 {"method": "fillna", "opt": "fill", "type": "str", "value": "其他"},
        "tar_qty":                                     {"method": "fillna", "opt": "mean", "type": "num"},
        "product_style_code_name":                     {"method": "fillna", "opt": "fill", "type": "str", "value": "其他"},
        "chinese_mix":                                 {"method": "fillna", "opt": "fill", "type": "str", "value": "否"},
        "sub_price_type_name":                         {"method": "fillna", "opt": "fill", "type": "str", "value": "普一类烟"},
    }
    
    ONEHOT = [
        "price_type_name",                             # 卷烟价类名称
        "gear_type_name",                              # 卷烟档位名称
        "category_type_name",                          # 卷烟品类名称
        "is_key_brand",                                # 是否重点品牌
        "is_high_level",                               # 是否高端烟
        "is_upscale_level",                            # 是否高端烟不含高价
        "is_high_price",                               # 是否高价烟
        "is_low_price",                                # 是否低价烟
        "is_low_tar",                                  # 是否低焦油烟
        "is_encourage",                                # 是否全国鼓励品牌
        "is_abnormity",                                # 是否异形包装
        "is_intake",                                   # 是否进口烟
        "is_short",                                    # 是否紧俏品牌
        "is_medium",                                   # 是否中支烟
        "is_shortbranch",                              # 是否短支烟
        "is_ordinary_price_type",                      # 是否普一类烟
        "source_type",                                 # 来源类型
        "product_style_code_name",                     # 包装类型名称
        "chinese_mix",                                 # 中式混合
        "sub_price_type_name",                         # 细分卷烟价类名称
    ]
    ONEHOT_CAT = {
        "price_type_name":                             ["一类烟", "二类烟", "三类烟", "四类烟", "五类烟", "无价类"],
        "gear_type_name":                              ["第一档位", "第二档位", "第三档位", "第四档位", "第五档位", "第六档位", "第七档位", "第八档位", "其他"],
        "category_type_name":                          ["第1品类", "第2品类", "第3品类", "第4品类", "第5品类", "第6品类", "第7品类", 
                                                        "第8品类", "第9品类", "第10品类", "第11品类", "第12品类", "第13品类", "其他"],
        "is_key_brand":                                ["是", "否"],
        "is_high_level":                               ["是", "否"],
        "is_upscale_level":                            ["是", "否"],
        "is_high_price":                               ["是", "否"],
        "is_low_price":                                ["是", "否"],
        "is_low_tar":                                  ["是", "否"],
        "is_encourage":                                ["是", "否"],
        "is_abnormity":                                ["是", "否"],
        "is_intake":                                   ["是", "否"],
        "is_short":                                    ["是", "否"],
        "is_medium":                                   ["是", "否"],
        "is_shortbranch":                              ["是", "否"],
        "is_ordinary_price_type":                      ["是", "否"],
        "source_type":                                 ["是", "否"],
        "product_style_code_name":                     ["条盒硬盒", "条包硬盒", "条盒软盒", "条包软盒", "铁盒", "其他"],
        "chinese_mix":                                 ["是", "否"],
        "sub_price_type_name":                         ["高端烟", "高价位烟", "普一类烟", "二类烟", "三类烟", "四类烟", "五类烟", "无价类"],
    }