We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e48fd7b commit 0a4d8c8Copy full SHA for 0a4d8c8
test/python/test_bcf.py
@@ -194,6 +194,19 @@ def test_binary_bcf(self):
194
# Check treatment effect prediction method
195
tau_hat = bcf_model.predict(X=X_test, Z=Z_test, terms="cate")
196
197
+ # Check that we can run BCF without propensities
198
+ bcf_model = BCFModel()
199
+ general_params = {"propensity_covariate": "none"}
200
+ bcf_model.sample(
201
+ X_train=X_train,
202
+ Z_train=Z_train,
203
+ y_train=y_train,
204
+ num_gfr=num_gfr,
205
+ num_burnin=num_burnin,
206
+ num_mcmc=num_mcmc,
207
+ general_params=general_params,
208
+ )
209
+
210
def test_continuous_univariate_bcf(self):
211
# RNG
212
random_seed = 101
0 commit comments