-
Notifications
You must be signed in to change notification settings - Fork 0
/
params.yaml
44 lines (39 loc) · 869 Bytes
/
params.yaml
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
44
randomforest:
n_estimators: 256
min_samples_split: 10
min_samples_leaf: 2
max_features: 'sqrt'
max_depth: 40
criterion: 'entropy'
oob_score: True
models:
GradientBoostingClassifier:
subsample: 0.7
n_estimators: 64
max_features: 'log2'
loss: 'exponential'
learning_rate: 0.1
criterion: 'friedman_mse'
XGBoostClassifier:
subsample: 0.6
n_estimators: 64
min_child_weight: 1
max_depth: 7
learning_rate: 0.1
CatBoostClassifier:
loss_function: 'CrossEntropy'
learning_rate: 0.1
iterations: 100
eval_metric: 'Logloss'
depth: 8
AdaBoostClassifier:
n_estimators: 16
learning_rate: 0.01
algorithm: 'SAMME.R'
RandomForestClassifier:
n_estimators: 256
min_samples_split: 10
min_samples_leaf: 2
max_features: 'sqrt'
max_depth: 40
criterion: 'entropy'