Commit 0db21565 authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

little fix

parent 5244ca0b
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -46,8 +46,8 @@ def load_gdb7(featurizer=None, split='random'):
  dataset_file = os.path.join(
      current_dir, "./gdb7.sdf")
  gdb7_tasks = ["u0_atom"]
  if featurizer == None:
    featurizer = dc.feat.CoulombMatrix(23)
  if featurizer is None:
    featurizer = dc.feat.CoulombMatrixEig(23)
  loader = dc.data.SDFLoader(tasks=gdb7_tasks, smiles_filed="smiles", 
                             mol_field="mol", featurizer=featurizer)
  dataset = loader.featurize(dataset_file)