apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-15 06:52:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

[splitkern] buf fix

Browse source
This commit is contained in:
Zhenwen Dai 2014-05-29 11:21:19 +01:00
parent 5d728f7715
commit f9291fe7da
1 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

8
GPy/kern/_src/splitKern.py
View file

@ -106,10 +106,10 @@ class SplitKern_cross(Kern):
k1 = self.kern.K(X,self.Xp)
k2 = self.kern.K(self.Xp,X2)
k3 = self.kern.K(self.Xp,self.Xp)
dL_dk1 = np.einsum('ij,j->i',dL_dK,k2[0])/k3
dL_dk2 = np.einsum('ij,i->j',dL_dK,k1[:,0])/k3
dL_dk3 = np.einsum('ij,ij->',dL_dK,-np.dot(k1,k2)/(k3*k3))
dL_dk1 = np.einsum('ij,j->i',dL_dK,k2.flat)/k3.flat
dL_dk2 = np.einsum('ij,i->j',dL_dK,k1.flat)/k3.flat
dL_dk3 = np.einsum('ij,ij->',dL_dK,-np.dot(k1,k2)/(k3.flat*k3.flat))
self.kern.update_gradients_full(dL_dk1[:,None],X,self.Xp)
grad1 = self.kern.gradient.copy()
self.kern.update_gradients_full(dL_dk2[None,:],self.Xp,X)