Commit 2a33962f authored by Vignesh's avatar Vignesh
Browse files

Minor Fixes

parent f1f94122

deepchem/molnet/load_function/bace_datasets.py

+6 −0
Original line number Diff line number Diff line
@@ -147,6 +147,12 @@ def load_bace_classification(featurizer='ECFP', split='random', reload=True): 


    return bace_tasks, (dataset, None, None), transformers



  splitters = {

      'index': deepchem.splits.IndexSplitter(),

      'random': deepchem.splits.RandomSplitter(),

      'scaffold': deepchem.splits.ScaffoldSplitter()

  }



  splitter = splitters[split]

  logger.info("About to split data using {} splitter".format(split))

  train, valid, test = splitter.train_valid_test_split(dataset)

deepchem/molnet/load_function/bbbc_datasets.py

+1 −0
Original line number Diff line number Diff line
@@ -139,6 +139,7 @@ def load_bbbc002(split='index', reload=True): 


  logger.info("About to split dataset with {} splitter.".format(split))

  train, valid, test = splitter.train_valid_test_split(dataset)

  all_dataset = (train, valid, test)

  transformers = []

  if reload:

    deepchem.utils.save.save_dataset_to_disk(save_dir, train, valid, test,

CRA Git | Maintained and supported by SUSTech CRA and CCSE