From 8dcd197436f3620ef63a8e25eaba38530e828f12 Mon Sep 17 00:00:00 2001 From: James Hensman Date: Tue, 14 May 2013 15:59:10 +0100 Subject: [PATCH] allowed GP models to plot multiple outputs (in 1D at least) --- GPy/models/GP.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/GPy/models/GP.py b/GPy/models/GP.py index e68ff68b..b518ce4e 100644 --- a/GPy/models/GP.py +++ b/GPy/models/GP.py @@ -253,8 +253,9 @@ class GP(model): Xnew, xmin, xmax = x_frame1D(Xu, plot_limits=plot_limits) m, var, lower, upper = self.predict(Xnew, which_parts=which_parts) - gpplot(Xnew, m, lower, upper) - pb.plot(Xu[which_data], self.likelihood.data[which_data], 'kx', mew=1.5) + for d in range(m.shape[1]): + gpplot(Xnew, m[:,d], lower[:,d], upper[:,d]) + pb.plot(Xu[which_data], self.likelihood.data[which_data,d], 'kx', mew=1.5) if self.has_uncertain_inputs: pb.errorbar(Xu[which_data, 0], self.likelihood.data[which_data, 0], xerr=2 * np.sqrt(self.X_variance[which_data, 0]),