Fixes the dimensions of the samples output

This commit is contained in:
Diego Torrejon 2018-03-21 18:00:13 -04:00
parent acc0d4f88d
commit 90c2912ace

View file

@ -592,7 +592,7 @@ class GP(Model):
if self.output_dim == 1: if self.output_dim == 1:
return sim_one_dim(m, v) return sim_one_dim(m, v)
else: else:
fsim = np.empty((self.output_dim, self.num_data, size)) fsim = np.empty((self.output_dim, len(X), size))
for d in range(self.output_dim): for d in range(self.output_dim):
if full_cov and v.ndim == 3: if full_cov and v.ndim == 3:
fsim[d] = sim_one_dim(m[:, d], v[:, :, d]) fsim[d] = sim_one_dim(m[:, d], v[:, :, d])