# 参数配置：分类器、数据集路径

class Config:
    
    DATA_PATH = "Dataset/KDDCUP99.csv"
    
    LABEL = [
        "X1",
        "X2",
        "X3",
        "X4",
        "X5",
        "X6",
        "X7",
        "X8",
        "X9"
    ]

    if_multi = True

    # Random Forest parameters
    rf_params = {
        'n_jobs': -1,
        'n_estimators': 500,
        'warm_start': True,
        'max_depth': 6,
        'min_samples_leaf': 2,
        'max_features' : 'sqrt',
        'verbose': 0
    }

    # Extra Trees parameters
    et_params = {
        'n_jobs': -1,
        'n_estimators':500,
        'max_depth': 8,
        'min_samples_leaf': 2,
        'verbose': 0
    }

    # AdaBoost parameters
    ada_params = {
        'n_estimators': 500,
        'learning_rate' : 0.75
    }

    # Gradient Boosting parameters
    gb_params = {
        'n_estimators': 500,
        'max_depth': 5,
        'min_samples_leaf': 2,
        'verbose': 0
    }

    # Support Vector Classifier parameters 
    svc_params = {
        'kernel': 'linear',
        'C': 0.025
    }

    mlp_params = {
        'alpha': 1.9, 
        'max_iter': 700
    }