easy-ml/enums/file_enum.py

# -*- coding: utf-8 -*-
"""
@author: yq
@time: 2024/11/14
@desc: 文件名枚举值
"""
from enum import Enum


class FileEnum(Enum):
    MLCFG = "mlcfg.json"
    OLCFG = "olcfg.json"
    FEATURE = "feature.csv"
    FEATURE_PKL = "feature.pkl"
    VAR_MAPPING = "var_mapping.csv"
    CARD = "card.csv"
    CARD_CFG = "card.cfg"
    COEF = "coef.json"
    MODEL = "model.pkl"
    MODEL_XGB = "xgb.bin"
    PMML = "model.pmml"