@@ -131,7 +131,7 @@ class PrePostFit(ExperimentalDesign):
131
131
... formula="actual ~ 0 + a + g",
132
132
... model=cp.pymc_models.WeightedSumFitter(
133
133
... sample_kwargs={
134
- ... "draws": 200 ,
134
+ ... "draws": 400 ,
135
135
... "target_accept": 0.95,
136
136
... "random_seed": seed,
137
137
... "progressbar": False
@@ -773,28 +773,14 @@ class RegressionDiscontinuity(ExperimentalDesign):
773
773
... formula="y ~ 1 + x + treated + x:treated",
774
774
... model=cp.pymc_models.LinearRegression(
775
775
... sample_kwargs={
776
- ... "draws": 200 ,
776
+ ... "draws": 100 ,
777
777
... "target_accept": 0.95,
778
778
... "random_seed": seed,
779
779
... "progressbar": False,
780
780
... },
781
781
... ),
782
782
... treatment_threshold=0.5,
783
783
... )
784
- >>> result.summary() # doctest: +NUMBER
785
- ============================Regression Discontinuity============================
786
- Formula: y ~ 1 + x + treated + x:treated
787
- Running variable: x
788
- Threshold on running variable: 0.5
789
- <BLANKLINE>
790
- Results:
791
- Discontinuity at threshold = 0.92
792
- Model coefficients:
793
- Intercept -0.9, 94% HDI [-1.0, -0.8]
794
- treated[T.True] 2.4, 94% HDI [1.7, 3.2]
795
- x 1.3, 94% HDI [1.1, 1.4]
796
- x:treated[T.True] -3.1, 94% HDI [-4.1, -2.0]
797
- sigma 0.3, 94% HDI [0.3, 0.4]
798
784
"""
799
785
800
786
def __init__ (
0 commit comments