Skip to content

Commit 05fccdd

Browse files
committed
remove redundant alpha parameter in model class
Signed-off-by: Nathaniel <[email protected]>
1 parent 548d723 commit 05fccdd

File tree

2 files changed

+681
-677
lines changed

2 files changed

+681
-677
lines changed

causalpy/pymc_models.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)