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())