b17c746750251ced6f1e46a7bce76a74e6fad47c,deepchem/molnet/load_function/pcba_datasets.py,,load_pcba_dataset,#Any#Any#Any#Any#,39

Before Change

  ]

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

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

  splitters = {

After Change

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

  transformers = [
      deepchem.trans.BalancingTransformer(transform_w=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)

Instances