add set_Z function

This commit is contained in:
Zhenwen Dai 2015-03-09 17:26:18 +00:00
parent 233c5ee8b4
commit c6cf0bc121

View file

@ -63,6 +63,15 @@ class SparseGP(GP):
def has_uncertain_inputs(self): def has_uncertain_inputs(self):
return isinstance(self.X, VariationalPosterior) return isinstance(self.X, VariationalPosterior)
def set_Z(self, Z):
self.update_model(False)
self.unlink_parameter(self.Z)
from ..core import Param
self.Z = Param('inducing inputs',Z)
self.link_parameter(self.Z, index=0)
self.update_model(True)
self._trigger_params_changed()
def parameters_changed(self): def parameters_changed(self):
self.posterior, self._log_marginal_likelihood, self.grad_dict = self.inference_method.inference(self.kern, self.X, self.Z, self.likelihood, self.Y, self.Y_metadata) self.posterior, self._log_marginal_likelihood, self.grad_dict = self.inference_method.inference(self.kern, self.X, self.Z, self.likelihood, self.Y, self.Y_metadata)