Merge pull request #148 from martinsch/normalization_fix

Normalization: avoid division by zero for constant feature dimensions
This commit is contained in:
James Hensman 2014-11-21 11:44:51 +00:00
commit 67ddbde119

View file

@ -29,6 +29,7 @@ class GPBase(Model):
if normalize_X: if normalize_X:
self._Xoffset = X.mean(0)[None, :] self._Xoffset = X.mean(0)[None, :]
self._Xscale = X.std(0)[None, :] self._Xscale = X.std(0)[None, :]
self._Xscale[np.where(self._Xscale==0)] = 1
self.X = (X.copy() - self._Xoffset) / self._Xscale self.X = (X.copy() - self._Xoffset) / self._Xscale
else: else:
self._Xoffset = np.zeros((1, self.input_dim)) self._Xoffset = np.zeros((1, self.input_dim))