def directwrap(Q, extra):
x = sp.array([Q])
s = sf
acq = PES.PESgain(G, Ga, Z, x, [[sp.NaN]], [s])
R = -acq / cost(x, s)
return (R, 0)
sp.array(mp0[0, :] + 2. * s).flatten(),
facecolor='lightblue',
edgecolor='lightblue')
a[0].plot(sup, mp0[0, :].flatten())
a[0].plot(Z[0, 0].flatten(), [0], 'r.')
s = sp.sqrt(Vp1[0, :])
a[1].fill_between(sup,
sp.array(mp1[0, :] - 2. * s).flatten(),
sp.array(mp1[0, :] + 2. * s).flatten(),
facecolor='lightblue',
edgecolor='lightblue')
a[1].plot(sup, mp1[0, :].flatten())
a[1].plot(Z[0, 1].flatten(), [0], 'r.')
H0 = PES.PESgain(G, Ga, Z, Xp0, Dp, [1e-3] * len(Dp))
a[0].twinx().plot(sup, H0.flatten(), 'r')
H1 = PES.PESgain(G, Ga, Z, Xp1, Dp, [1e-3] * len(Dp))
a[1].twinx().plot(sup, H1.flatten(), 'r')
##before
[mp0_, Vp0_] = G.infer_diag_post(Xp0, Dp)
[mp1_, Vp1_] = G.infer_diag_post(Xp1, Dp)
f, a = plt.subplots(d)
s = sp.sqrt(Vp0_[0, :])
a[0].fill_between(sup,
sp.array(mp0_[0, :] - 2. * s).flatten(),
sp.array(mp0_[0, :] + 2. * s).flatten(),
facecolor='lightblue',
edgecolor='lightblue')
a[0].plot(sup, mp0_[0, :].flatten())
