Skip to content

Commit 42e9bcf

Browse files
Junpeng Laotwiecki
authored andcommitted
fix #2108 (#2110)
* fix #2108 * clearner solution for fixing #2108 * bug fixed
1 parent 6279ed4 commit 42e9bcf

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pymc3/sampling.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -573,7 +573,8 @@ def init_nuts(init='ADVI', njobs=1, n_init=500000, model=None,
573573
progressbar=progressbar
574574
) # type: pm.MeanField
575575
start = approx.sample(draws=njobs)
576-
cov = approx.cov.eval()
576+
stds = approx.gbij.rmap(np.diag(approx.cov.eval()))
577+
cov = model.dict_to_array(stds)
577578
if njobs == 1:
578579
start = start[0]
579580
elif init == 'advi_map':

0 commit comments

Comments
 (0)