little fix

kyscg · Feb 1, 2017 · 0db2156 · 0db2156
1 parent 5244ca0
commit 0db2156
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/gdb7/gdb7_datasets.py b/examples/gdb7/gdb7_datasets.py
@@ -46,9 +46,9 @@ 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)
-  loader = dc.data.SDFLoader(tasks=gdb7_tasks, smiles_filed="smiles",
+  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)