Commit ccd934f1 authored by miaecle's avatar miaecle
Browse files

little change

parent 49bc3cec
Loading
Loading
Loading
Loading
+5 −4
Original line number Diff line number Diff line
@@ -10,7 +10,7 @@ import numpy as np
import shutil
import deepchem as dc

def load_gdb7(featurizer=None, split='index'):
def load_gdb7(featurizer=None, split='random'):
  """Load gdb7 datasets."""
  # Featurize gdb7 dataset
  print("About to featurize gdb7 dataset.")
@@ -39,4 +39,5 @@ def load_gdb7(featurizer=None, split='index'):
               'random': dc.splits.RandomSplitter()}
  splitter = splitters[split]
  train, valid, test = splitter.train_valid_test_split(dataset)
 
  return gdb7_tasks, (train, valid, test), transformers