added keywords to predict in warped gps because they are used in the plotting method

This commit is contained in:
beckdaniel 2015-07-30 17:19:16 +01:00
parent 7a7d563576
commit 45321f536b

View file

@ -69,7 +69,7 @@ class WarpedGP(GP):
def plot_warping(self): def plot_warping(self):
self.warping_function.plot(self.Y_untransformed.min(), self.Y_untransformed.max()) self.warping_function.plot(self.Y_untransformed.min(), self.Y_untransformed.max())
def predict(self, Xnew, which_parts='all', pred_init=None): def predict(self, Xnew, which_parts='all', pred_init=None, Y_metadata=None, full_cov=False):
# normalize X values # normalize X values
# Xnew = (Xnew.copy() - self._Xoffset) / self._Xscale # Xnew = (Xnew.copy() - self._Xoffset) / self._Xscale
mu, var = GP._raw_predict(self, Xnew) mu, var = GP._raw_predict(self, Xnew)