File tree Expand file tree Collapse file tree 1 file changed +13
-9
lines changed Expand file tree Collapse file tree 1 file changed +13
-9
lines changed Original file line number Diff line number Diff line change @@ -7,7 +7,7 @@ simulation_parameters:
7
7
n_jobs : -2
8
8
9
9
dgp_parameters :
10
- n_obs : [1000 ] # Sample size
10
+ n_obs : [2000 ] # Sample size
11
11
fuzzy : [True]
12
12
cutoff : [0.0]
13
13
@@ -16,14 +16,16 @@ learner_definitions:
16
16
lgbmr : &lgbmr
17
17
name : " LGBM Regr."
18
18
params :
19
- n_estimators : 100
20
- learning_rate : 0.05
19
+ n_estimators : 200
20
+ learning_rate : 0.02
21
+ max_depth : 5
21
22
22
23
lgbmc : &lgbmc
23
24
name : " LGBM Clas."
24
25
params :
25
- n_estimators : 100
26
- learning_rate : 0.05
26
+ n_estimators : 200
27
+ learning_rate : 0.02
28
+ max_depth : 5
27
29
28
30
global_linear : &global_linear
29
31
name : " Global Linear"
@@ -40,14 +42,16 @@ learner_definitions:
40
42
stacked_reg : &stacked_reg
41
43
name : " Stacked Regr."
42
44
params :
43
- n_estimators : 100
44
- learning_rate : 0.05
45
+ n_estimators : 200
46
+ learning_rate : 0.02
47
+ max_depth : 5
45
48
46
49
stacked_cls : &stacked_cls
47
50
name : " Stacked Clas."
48
51
params :
49
- n_estimators : 100
50
- learning_rate : 0.05
52
+ n_estimators : 200
53
+ learning_rate : 0.02
54
+ max_depth : 5
51
55
52
56
dml_parameters :
53
57
fs_specification : ["cutoff", "cutoff and score", "interacted cutoff and score"]
You can’t perform that action at this time.
0 commit comments