-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathtest-3.py
More file actions
34 lines (32 loc) · 1002 Bytes
/
test-3.py
File metadata and controls
34 lines (32 loc) · 1002 Bytes
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
params = (
{
'criterion': ['entropy', 'gini'],
'bootstrap': [True, False],
'max_depth': [5, 10, 20, 30, 50],
'max_features': ['auto', 'sqrt'],
'min_samples_leaf': [5, 10, 20, 30, 50],
'min_samples_split': [5, 10, 30, 50, 70],
'n_estimators': [100, 200, 300, 400]
}
,
{
"learning_rate": [0.05, 0.15],
"max_depth": [5, 10, 20, 30, 50, 70],
"min_child_weight": [5, 10, 20, 50, 100],
"gamma": [0.0, 0.1, 0.2],
"colsample_bytree": [0.2, 0.3, 0.5],
"n_estimators": [100, 200, 300, 400]
}
,
{
'random_state': [42],
'objective': ['binary'],
'class_weight': ['balanced', None],
'min_split_gain': [0, 0.005, 0.01, 0.02, 0.05, 0.1],
'max_depth': [5, 10, 20, 30, 50],
'n_estimators': [100, 200, 300, 400, 600, 800],
'reg_alpha': [0, 0.1, 0.2, 0.3, 0.5],
'reg_lambda': [0, 0.1, 0.2, 0.3, 0.5]
}
)
print(params[0])