File tree Expand file tree Collapse file tree 2 files changed +681
-677
lines changed
Expand file tree Collapse file tree 2 files changed +681
-677
lines changed Original file line number Diff line number Diff line change @@ -712,11 +712,7 @@ class initialisation.
712712 chosen = np .random .choice (range (propensity_scores .shape [1 ]))
713713 p = propensity_scores [:, chosen ].values
714714
715- alpha_outcome = pm .Normal (
716- "a_outcome" , priors ["a_outcome" ][0 ], priors ["a_outcome" ][1 ]
717- )
718-
719- mu_outcome = alpha_outcome + pm .math .dot (X_data_outcome , beta ) + beta_ps * p
715+ mu_outcome = pm .math .dot (X_data_outcome , beta ) + beta_ps * p
720716
721717 if spline_component :
722718 beta_ps_spline = pm .Normal (
@@ -733,9 +729,7 @@ class initialisation.
733729 splines_summed = pm .Deterministic (
734730 "spline_features" , pm .math .dot (B_f , beta_ps_spline .T )
735731 )
736- mu_outcome = (
737- alpha_outcome + pm .math .dot (X_data_outcome , beta ) + splines_summed
738- )
732+ mu_outcome = pm .math .dot (X_data_outcome , beta ) + splines_summed
739733
740734 sigma = pm .HalfNormal ("sigma" , priors ["sigma" ])
741735
You can’t perform that action at this time.
0 commit comments