mirror of
https://github.com/SheffieldML/GPy.git
synced 2026-05-09 12:02:38 +02:00
passing psi statistics
This commit is contained in:
parent
1a2b194aa2
commit
c83f56723e
1 changed files with 16 additions and 10 deletions
|
|
@ -167,21 +167,27 @@ class SparseGPMiniBatch(SparseGP):
|
||||||
dL_dKmm = full_values['dL_dKmm']
|
dL_dKmm = full_values['dL_dKmm']
|
||||||
self.kern.update_gradients_full(dL_dKmm, self.Z, None)
|
self.kern.update_gradients_full(dL_dKmm, self.Z, None)
|
||||||
full_values['kerngrad'] = self.kern.gradient.copy()
|
full_values['kerngrad'] = self.kern.gradient.copy()
|
||||||
self.kern.update_gradients_expectations(variational_posterior=self.X,
|
self.kern.update_gradients_expectations(
|
||||||
Z=self.Z,
|
variational_posterior=self.X,
|
||||||
dL_dpsi0=full_values['dL_dpsi0'],
|
Z=self.Z, dL_dpsi0=full_values['dL_dpsi0'],
|
||||||
dL_dpsi1=full_values['dL_dpsi1'],
|
dL_dpsi1=full_values['dL_dpsi1'],
|
||||||
dL_dpsi2=full_values['dL_dpsi2'])
|
dL_dpsi2=full_values['dL_dpsi2'],
|
||||||
|
psi0=self.psi0, psi1=self.psi1, psi2=self.psi2)
|
||||||
|
#self.kern.update_gradients_expectations(variational_posterior=self.X,
|
||||||
|
#Z=self.Z,
|
||||||
|
#dL_dpsi0=full_values['dL_dpsi0'],
|
||||||
|
#dL_dpsi1=full_values['dL_dpsi1'],
|
||||||
|
#dL_dpsi2=full_values['dL_dpsi2'])
|
||||||
full_values['kerngrad'] += self.kern.gradient
|
full_values['kerngrad'] += self.kern.gradient
|
||||||
|
|
||||||
#gradients wrt Z
|
#gradients wrt Z
|
||||||
full_values['Zgrad'] = self.kern.gradients_X(dL_dKmm, self.Z)
|
full_values['Zgrad'] = self.kern.gradients_X(dL_dKmm, self.Z)
|
||||||
full_values['Zgrad'] += self.kern.gradients_Z_expectations(
|
full_values['Zgrad'] += self.kern.gradients_Z_expectations(
|
||||||
full_values['dL_dpsi0'],
|
variational_posterior=self.X,
|
||||||
full_values['dL_dpsi1'],
|
Z=self.Z, dL_dpsi0=full_values['dL_dpsi0'],
|
||||||
full_values['dL_dpsi2'],
|
dL_dpsi1=full_values['dL_dpsi1'],
|
||||||
Z=self.Z,
|
dL_dpsi2=full_values['dL_dpsi2'],
|
||||||
variational_posterior=self.X)
|
psi0=self.psi0, psi1=self.psi1, psi2=self.psi2)
|
||||||
else:
|
else:
|
||||||
#gradients wrt kernel
|
#gradients wrt kernel
|
||||||
self.kern.update_gradients_diag(full_values['dL_dKdiag'], self.X)
|
self.kern.update_gradients_diag(full_values['dL_dKdiag'], self.X)
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue