Skip to content

Commit eef4c78

Browse files
committed
update fuzzy config
1 parent b5fe71f commit eef4c78

File tree

1 file changed

+13
-9
lines changed

1 file changed

+13
-9
lines changed

scripts/rdd/rdd_fuzzy_config.yml

Lines changed: 13 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ simulation_parameters:
77
n_jobs: -2
88

99
dgp_parameters:
10-
n_obs: [1000] # Sample size
10+
n_obs: [2000] # Sample size
1111
fuzzy: [True]
1212
cutoff: [0.0]
1313

@@ -16,14 +16,16 @@ learner_definitions:
1616
lgbmr: &lgbmr
1717
name: "LGBM Regr."
1818
params:
19-
n_estimators: 100
20-
learning_rate: 0.05
19+
n_estimators: 200
20+
learning_rate: 0.02
21+
max_depth: 5
2122

2223
lgbmc: &lgbmc
2324
name: "LGBM Clas."
2425
params:
25-
n_estimators: 100
26-
learning_rate: 0.05
26+
n_estimators: 200
27+
learning_rate: 0.02
28+
max_depth: 5
2729

2830
global_linear: &global_linear
2931
name: "Global Linear"
@@ -40,14 +42,16 @@ learner_definitions:
4042
stacked_reg: &stacked_reg
4143
name: "Stacked Regr."
4244
params:
43-
n_estimators: 100
44-
learning_rate: 0.05
45+
n_estimators: 200
46+
learning_rate: 0.02
47+
max_depth: 5
4548

4649
stacked_cls: &stacked_cls
4750
name: "Stacked Clas."
4851
params:
49-
n_estimators: 100
50-
learning_rate: 0.05
52+
n_estimators: 200
53+
learning_rate: 0.02
54+
max_depth: 5
5155

5256
dml_parameters:
5357
fs_specification: ["cutoff", "cutoff and score", "interacted cutoff and score"]

0 commit comments

Comments
 (0)