Updates for eq_ode1 and eq_ode2 kernels

This commit is contained in:
cdguarnizo 2016-05-23 01:28:01 -05:00
parent f844143353
commit 52fb928dff
3 changed files with 105 additions and 29 deletions

View file

@ -58,7 +58,7 @@ class VarDTC_minibatch_IBPLFM(VarDTC_minibatch):
else:
b = beta
psi0 = kern.Kdiag(X_slice) #Kff^q
psi0 = kern._Kdiag(X_slice) #Kff^q
psi1 = kern.K(X_slice, Z) #Kfu
indX = X_slice.values
@ -223,7 +223,7 @@ class VarDTC_minibatch_IBPLFM(VarDTC_minibatch):
Y_slice = YYT_factor[n_start:n_end]
X_slice = X[n_start:n_end]
psi0 = kern.Kdiag(X_slice) #Kffdiag
psi0 = kern._Kdiag(X_slice) #Kffdiag
psi1 = kern.K(X_slice, Z) #Kfu
betapsi1 = np.einsum('n,nm->nm', beta, psi1)