easy-ml/enums/file_enum.py

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


class FileEnum(Enum):
    ML_CFG = "ml_cfg.json"
    OL_CFG = "ol_cfg.json"
    FEATURE_CSV = "feature.csv"
    FEATURE_PKL = "feature.pkl"
    VAR_MAPPING_CSV = "var_mapping.csv"
    CARD_CSV = "card.csv"
    CARD_CFG = "card_cfg.json"
    COEF = "coef.json"
    PIPELINE_XGB = "xgb.pipeline"
    MODEL_XGB = "xgb.bin"
    PMML_XGB = "xgb.pmml"
    TEST_CASE_CSV = "test_case.csv"