Commit 31e826ca authored by Joseph Gomes's avatar Joseph Gomes
Browse files

Fixed bug for NormalizationTransformer on singletask datasets

parent 83c20492
Loading
Loading
Loading
Loading
+1 −0
Original line number Diff line number Diff line
@@ -104,6 +104,7 @@ class NormalizationTransformer(Transformer):
      y_means, y_stds = dataset.get_statistics(X_stats=False, y_stats=True)
      self.y_means = y_means 
      # Control for pathological case with no variance.
      y_stds = np.array(y_stds)
      y_stds[y_stds == 0] = 1.
      self.y_stds = y_stds
    self.transform_gradients = transform_gradients