Commit e9eb598a authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #361 from joegomes/singletask_fixes 

NormalizationTransformer on single task datasets
parents a37c108a 1664886d

deepchem/trans/transformers.py

+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

deepchem/utils/save.py

+2 −0
Original line number Diff line number Diff line
@@ -108,6 +108,8 @@ def load_from_disk(filename): 
    except KeyError:

      # Try older joblib version for legacy files.

      return old_joblib.load(filename)

    except ValueError:

      return old_joblib.load(filename)

  elif os.path.splitext(name)[1] == ".csv":

    # First line of user-specified CSV *must* be header.

    df = pd.read_csv(filename, header=0)

CRA Git | Maintained and supported by SUSTech CRA and CCSE