Commit 860f9eff authored by leswing's avatar leswing
Browse files

Do Core

parent 699c473b
Loading
Loading
Loading
Loading
+1 −1
Original line number Original line Diff line number Diff line
@@ -142,7 +142,7 @@ def compute_pdbbind_features(grid_featurizer, pdb_subdir, pdb_code):
  return features
  return features




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