Skip to content

Commit 150cedc

Browse files
author
Junpeng Lao
committed
fix issue #2247
*cast value to float * might not be an issue any more after #2233
1 parent 6f58dbf commit 150cedc

File tree

1 file changed

+6
-7
lines changed

1 file changed

+6
-7
lines changed

pymc3/examples/lasso_missing.py

Lines changed: 6 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,32 +16,31 @@
1616
with pm.Model() as model:
1717

1818
# Impute missing values
19-
sib_mean = pm.Exponential('sib_mean', 1)
19+
sib_mean = pm.Exponential('sib_mean', 1.)
2020
siblings_imp = pm.Poisson('siblings_imp', sib_mean,
2121
observed=masked_values(siblings, value=-999))
2222

23-
p_disab = pm.Beta('p_disab', 1, 1)
23+
p_disab = pm.Beta('p_disab', 1., 1.)
2424
disability_imp = pm.Bernoulli(
2525
'disability_imp', p_disab, observed=masked_values(disability, value=-999))
2626

27-
p_mother = pm.Beta('p_mother', 1, 1)
27+
p_mother = pm.Beta('p_mother', 1., 1.)
2828
mother_imp = pm.Bernoulli('mother_imp', p_mother,
2929
observed=masked_values(mother_hs, value=-999))
3030

31-
s = pm.HalfCauchy('s', 5, testval=5)
32-
beta = pm.Laplace('beta', 0, 100, shape=7, testval=.1)
31+
s = pm.HalfCauchy('s', 5., testval=5)
32+
beta = pm.Laplace('beta', 0., 100., shape=7, testval=.1)
3333

3434
expected_score = (beta[0] + beta[1] * male + beta[2] * siblings_imp + beta[3] * disability_imp +
3535
beta[4] * age + beta[5] * mother_imp + beta[6] * early_ident)
3636

3737
observed_score = pm.Normal(
38-
'observed_score', expected_score, s, observed=score)
38+
'observed_score', expected_score, tau=s, observed=score)
3939

4040

4141
with model:
4242
start = pm.find_MAP()
4343
step1 = pm.NUTS([beta, s, p_disab, p_mother, sib_mean], scaling=start)
44-
4544
step2 = pm.Metropolis([mother_imp.missing_values,
4645
disability_imp.missing_values,
4746
siblings_imp.missing_values])

0 commit comments

Comments
 (0)