Commit 03ca10f2 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Adding None split

parent fff925b1
Loading
Loading
Loading
Loading
+3 −0
Original line number Diff line number Diff line
@@ -47,6 +47,9 @@ def load_sweet(featurizer='ECFP', split='index', reload=True, frac_train=.8):
  for transformer in transformers:
      dataset = transformer.transform(dataset)

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

  splitters = {
      'index': dc.splits.IndexSplitter(),
      'random': dc.splits.RandomSplitter(),