@@ -131,7 +131,7 @@ class PrePostFit(ExperimentalDesign):
131131 ... formula="actual ~ 0 + a + g",
132132 ... model=cp.pymc_models.WeightedSumFitter(
133133 ... sample_kwargs={
134- ... "draws": 200 ,
134+ ... "draws": 400 ,
135135 ... "target_accept": 0.95,
136136 ... "random_seed": seed,
137137 ... "progressbar": False
@@ -773,28 +773,14 @@ class RegressionDiscontinuity(ExperimentalDesign):
773773 ... formula="y ~ 1 + x + treated + x:treated",
774774 ... model=cp.pymc_models.LinearRegression(
775775 ... sample_kwargs={
776- ... "draws": 200 ,
776+ ... "draws": 100 ,
777777 ... "target_accept": 0.95,
778778 ... "random_seed": seed,
779779 ... "progressbar": False,
780780 ... },
781781 ... ),
782782 ... treatment_threshold=0.5,
783783 ... )
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]
798784 """
799785
800786 def __init__ (
0 commit comments