Skip to content

Commit 429fb3c

Browse files
author
Junpeng Lao
committed
Fix lasso_missing.py example
close #2417 * error caused by using mask on non missing array siblings * use BinaryGibbsMetropolis instead of Metropolis (BinaryGibbsMetropolis was auto assigned, also metropolis doesnt work as trace stuck at 0)
1 parent 875fd7e commit 429fb3c

File tree

1 file changed

+3
-4
lines changed

1 file changed

+3
-4
lines changed

pymc3/examples/lasso_missing.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@
1717
# Impute missing values
1818
sib_mean = pm.Exponential('sib_mean', 1.)
1919
siblings_imp = pm.Poisson('siblings_imp', sib_mean,
20-
observed=masked_values(siblings, value=-999))
20+
observed=siblings)
2121

2222
p_disab = pm.Beta('p_disab', 1., 1.)
2323
disability_imp = pm.Bernoulli(
@@ -40,9 +40,8 @@
4040
with model:
4141
start = pm.find_MAP()
4242
step1 = pm.NUTS([beta, s, p_disab, p_mother, sib_mean], scaling=start)
43-
step2 = pm.Metropolis([mother_imp.missing_values,
44-
disability_imp.missing_values,
45-
siblings_imp.missing_values])
43+
step2 = pm.BinaryGibbsMetropolis([mother_imp.missing_values,
44+
disability_imp.missing_values])
4645

4746

4847
def run(n=5000):

0 commit comments

Comments
 (0)