Python中如何获取category_encoder生成的类别与编码值映射关系?
获取WOEEncoder类别与WoE值的对应关系
category_encoders库的WOEEncoder拟合后返回的mapping属性本身就包含完整的类别- WoE对应关系,结构如下:
encoder.mapping是一个列表,长度等于你拟合时传入的待编码特征数量- 列表中的每个元素是对应单个特征的映射字典,包含两个核心字段:
col:当前映射对应的特征名称mapping:一个Pandas Series对象,Series的索引就是原始类别值,Series的值就是对应的WoE值
你可以通过以下代码直接输出所有特征的完整映射关系:
# 遍历输出所有特征的类别-WoE对应规则 for feature_map in encoder.mapping: print(f"特征名称:{feature_map['col']}") print("类别 -> 对应WoE值") # 直接输出Series,索引为类别、值为WoE print(feature_map['mapping']) print("-" * 40)
如果需要单独提取某一个特征的映射为Python字典,方便查询某个类别的对应WoE值,可以用以下代码:
# 替换为你要查询的特征名 target_col = "你的待查询特征名" for feature_map in encoder.mapping: if feature_map["col"] == target_col: # 把Series转成Python字典,key为类别、value为WoE值 cate_to_woe = feature_map["mapping"].to_dict() break # 直接查询指定类别的WoE值 print(cate_to_woe["A"]) # 输出示例:0.2
你之前直接打印encoder.mapping没有对应上类别和值,是因为你只关注了Series的数值部分,忽略了默认输出时左侧的索引就是原始类别,转成字典后就可以直接看到键值对应关系。
内容的提问来源于stack exchange,提问作者eduardokapp
相关产品推荐
相关产品推荐

