ss_mrd with parameter tied

This commit is contained in:
Zhenwen Dai 2014-06-27 17:25:08 +01:00
parent 35eb0a015f
commit 2c88528ebd

View file

@ -20,3 +20,12 @@ class SSMRD(Model):
self.add_parameters(*(self.models))
self.updates = True
[[self.models[j].X.mean.flat[i:i+1].tie('mean_'+str(i)) for j in xrange(len(self.models))] for i in xrange(self.models[0].X.mean.size)]
[[self.models[j].X.variance.flat[i:i+1].tie('var_'+str(i)) for j in xrange(len(self.models))] for i in xrange(self.models[0].X.variance.size)]
def parameters_changed(self):
super(SSMRD, self).parameters_changed()
self._log_marginal_likelihood = sum([m._log_marginal_likelihood for m in self.models])
def log_likelihood(self):
return self._log_marginal_likelihood