b17c746750251ced6f1e46a7bce76a74e6fad47c,deepchem/molnet/load_function/nci_datasets.py,,load_nci,#Any#Any#Any#Any#,16

Before Change

      deepchem.trans.NormalizationTransformer(
          transform_y=True, dataset=dataset)
  ]
  for transformer in transformers:
    dataset = transformer.transform(dataset)

  if split == None:
    return all_nci_tasks, (dataset, None, None), transformers

  splitters = {

After Change

  logger.info("About to split data with {} splitter.".format(splitter))
  train, valid, test = splitter.train_valid_test_split(dataset)

  transformers = [
      deepchem.trans.NormalizationTransformer(transform_y=True, dataset=train)
  ]

  logger.info("About to transform dataset.")
  for transformer in transformers:
    train = transformer.transform(train)
    valid = transformer.transform(valid)
    test = transformer.transform(test)

  if reload:
    deepchem.utils.save.save_dataset_to_disk(save_dir, train, valid, test,
                                             transformers)
  return all_nci_tasks, (train, valid, test), transformers

Instances