easy-ml/enums/context_enum.py

# -*- coding: utf-8 -*-
"""
@author: yq
@time: 2024/11/14
@desc: 特征筛选策略枚举值
"""
from enum import Enum


class ContextEnum(Enum):
    # 分箱信息
    WOEBIN = "woebin"

    FILTER_FAST = "filter_fast"
    FILTER_NUMERIC = "filter_numeric"
    FILTER_CORR = "filter_corr"
    FILTER_VIF = "filter_vif"
    FILTER_IVTOP = "filter_ivtop"

    XGB_COLUMNS_SELECTED = "xgb_columns_selected"
    XGB_COLUMNS_NUM = "xgb_columns_num"
    XGB_POINTS = "xgb_points"