Unverified Commit 795aa5b9 authored by Karl Leswing's avatar Karl Leswing Committed by GitHub
Browse files

Merge pull request #1131 from nitinprakash96/master 

Return entire dataset when split==None
parents 45190437 33115c21

deepchem/molnet/load_function/delaney_datasets.py

+14 −11
Original line number Diff line number Diff line
@@ -53,6 +53,9 @@ def load_delaney(featurizer='ECFP', split='index', reload=True): 
  for transformer in transformers:

    dataset = transformer.transform(dataset)



  if split == None:

    return delaney_tasks, (dataset, None, None), transformers

  else:

    splitters = {

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

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

CRA Git | Maintained and supported by SUSTech CRA and CCSE