File size: 339 Bytes
49a060f
 
 
 
 
 
 
 
 
 
 
 
 
9e6c24e
49a060f
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
import os


DATA_ROOT = os.path.abspath(os.path.join(__file__, "../..", "data"))

# MODEL
SUPPORT_MODEL = (
    "LogisticRegression",
    "DecisionTreeClassifier",
    "MLPClassifier",
    "RandomForestClassifier",
    "LGBMClassifier",
    "XGBClassifier",
    "GradientBoostingClassifier",
)
DEFAULT_MODEL = "GradientBoostingClassifier"