Commit ab936d7b authored by ZHENQIN WU's avatar ZHENQIN WU
Browse files

little fix

parent 0db21565

examples/gdb7/gdb7_datasets.py

+1 −1
Original line number Diff line number Diff line
@@ -48,7 +48,7 @@ def load_gdb7(featurizer=None, split='random'): 
  gdb7_tasks = ["u0_atom"]

  if featurizer is None:

    featurizer = dc.feat.CoulombMatrixEig(23)

  loader = dc.data.SDFLoader(tasks=gdb7_tasks, smiles_filed="smiles", 

  loader = dc.data.SDFLoader(tasks=gdb7_tasks, smiles_field="smiles", 

                             mol_field="mol", featurizer=featurizer)

  dataset = loader.featurize(dataset_file)

CRA Git | Maintained and supported by SUSTech CRA and CCSE