From 2c88528ebd96ecb8ce53a9614bdfc3d0a41ccc61 Mon Sep 17 00:00:00 2001 From: Zhenwen Dai Date: Fri, 27 Jun 2014 17:25:08 +0100 Subject: [PATCH] ss_mrd with parameter tied --- GPy/models/ss_mrd.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/GPy/models/ss_mrd.py b/GPy/models/ss_mrd.py index f959eb59..612ea350 100644 --- a/GPy/models/ss_mrd.py +++ b/GPy/models/ss_mrd.py @@ -19,4 +19,13 @@ class SSMRD(Model): name='model_'+str(i)) for i,y in enumerate(Ylist)] 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 \ No newline at end of file