Commit b148e8ad authored by Bharath Ramsundar's avatar Bharath Ramsundar Committed by GitHub
Browse files

Merge pull request #661 from lilleswing/660-pdb_rf

Revert default args for load_pdbbind_grid
parents 0f7e587b fd7a6c87
Loading
Loading
Loading
Loading
+1 −2
Original line number Diff line number Diff line
@@ -185,8 +185,7 @@ 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