[gradxx] not working with X,X...

This commit is contained in:
mzwiessele 2016-05-06 16:02:53 +01:00
parent b16d57f560
commit 17bfccb457
4 changed files with 41 additions and 23 deletions

View file

@ -419,7 +419,7 @@ class GP(Model):
mumuT = np.einsum('iqd,ipd->iqp', mu_jac, mu_jac)
Sigma = np.zeros(mumuT.shape)
if var_jac.ndim == 4: # Missing data
Sigma[(slice(None), )+np.diag_indices(Xnew.shape[1], 2)] = var_jac.sum(-1)
Sigma = var_jac.sum(-1)
else:
Sigma = self.output_dim*var_jac