Commit b556ddda authored by leswing's avatar leswing
Browse files

Change load defaults

parent 9569295e
Loading
Loading
Loading
Loading
+2 −2
Original line number Diff line number Diff line
@@ -185,8 +185,8 @@ def compute_pdbbind_features(grid_featurizer, pdb_subdir, pdb_code):
  return features


def load_pdbbind_grid(split="index", featurizer="atomic_conv",
                      subset="refined"):
def load_pdbbind_grid(split="index", featurizer="grid",
                      subset="core"):
  """Load PDBBind datasets. Does not do train/test split"""
  dataset, tasks = featurize_pdbbind(feat=featurizer, subset=subset)
  dataset.w