Skip to content

Commit 56c32bf

Browse files
committed
update cvar setting to allow more overfitting
1 parent 2f8b1c6 commit 56c32bf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

doubleml/irm/tests/test_cvar_tune_ml_models.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,10 @@
1616
@pytest.mark.parametrize("sampler_name,optuna_sampler", _SAMPLER_CASES, ids=[case[0] for case in _SAMPLER_CASES])
1717
def test_doubleml_cvar_optuna_tune(sampler_name, optuna_sampler):
1818
np.random.seed(3145)
19-
dml_data = make_irm_data(n_obs=500, dim_x=5)
19+
dml_data = make_irm_data(n_obs=200, dim_x=5)
2020

21-
ml_g = DecisionTreeRegressor(random_state=321, max_depth=None, min_samples_split=2)
22-
ml_m = DecisionTreeClassifier(random_state=654, max_depth=None, min_samples_split=2)
21+
ml_g = DecisionTreeRegressor(random_state=321, max_depth=None, min_samples_split=2, min_samples_leaf=1)
22+
ml_m = DecisionTreeClassifier(random_state=654, max_depth=None, min_samples_split=2, min_samples_leaf=1)
2323

2424
dml_cvar = dml.DoubleMLCVAR(dml_data, ml_g=ml_g, ml_m=ml_m, n_folds=2)
2525
dml_cvar.fit()

0 commit comments

Comments
 (0)