-
Notifications
You must be signed in to change notification settings - Fork 0
/
lists.py
43 lines (35 loc) · 1002 Bytes
/
lists.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
def task_names():
return ['Classification', 'Regression', 'Clustering']
def model_lists(task_name):
if task_name == 'Classification':
return [
"RandomForest",
"DecisionTree",
"XGBoost",
"CatBoost",
"Naive Bayes",
"Logistic Regression",
"Support Vector Machines"
]
elif task_name == "Regression":
return [
"Linear Regression",
"ElasticNet",
"Stochastic Gradient Descent",
"RandomForest",
"DecisionTree",
"XGBoost",
"CatBoost",
"Support Vector Machines",
]
elif task_name == "Clustering":
return [
"KMeans",
"Hierarchical"
]
else:
return [None]
def transformation_list():
return ['Standardization', "Normalization"]
def encoding_list():
return ['LabelEncoding', "OneHotEncoding"]