clf_model_abbreviations = [
    "rf_clf",
    "et_clf",
    "gb_clf",
    "dt_clf",
    "logistic",
    "svc",
    "knn_clf",
    "nn",
    "ada_clf",
    "vc_clf",
]

clf_models = [
    "RandomForestClassifier",
    "ExtraTreesClassifier",
    "GradientBoostingClassifier",
    "DecisionTreeClassifier",
    "LogisticRegression",
    "SVC",
    "KNeighborsClassifier",
    "MLPClassifier",
    "AdaBoostClassifier",
    "VotingClassifier",
]

clf_metrics = [
    "accuracy",
    "f1",
    "precision",
    "recall",
    "roc_auc",
    "precision_recall_auc",
]

regress_model_abbreviations = [
    "rf_regress",
    "et_regress",
    "gb_regress",
    "dt_regress",
    "linear",
    "lasso",
    "elastic",
    "svr",
    "knn_regress",
    "ada_regress",
    "vc_regress",
]

regress_models = [
    "RandomForestRegressor",
    "ExtraTreesRegressor",
    "GradientBoostingRegressor",
    "DecisionTreeRegressor",
    "LinearRegression",
    "Lasso",
    "ElasticNet",
    "SVR",
    "KNeighborsRegressor",
    "AdaBoostRegressor",
    "VotingRegressor",
]

regress_metrics = ["mse", "rmse", "mae", "mape", "r2"]
