[white hetero] additional check for prediction

This commit is contained in:
mzwiessele 2016-03-07 15:31:20 +00:00
parent 90f29405ba
commit c72d085e2b

View file

@ -107,7 +107,7 @@ class WhiteHeteroscedastic(Static):
if X2 is None and X.shape[0] == self.variance.shape[0]: if X2 is None and X.shape[0] == self.variance.shape[0]:
return np.eye(X.shape[0]) * self.variance return np.eye(X.shape[0]) * self.variance
else: else:
return np.zeros((X.shape[0], X2.shape[0])) return 0.
def psi2(self, Z, variational_posterior): def psi2(self, Z, variational_posterior):
return np.zeros((Z.shape[0], Z.shape[0]), dtype=np.float64) return np.zeros((Z.shape[0], Z.shape[0]), dtype=np.float64)