Commit 612c5f80 authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Saving changes

parent cbf3580f
Loading
Loading
Loading
Loading
+1 −1
Original line number Diff line number Diff line
@@ -19,7 +19,7 @@ from binding_pocket_datasets import load_pdbbind_pockets
np.random.seed(123)

split = "random"
subset = "refined"
subset = "full"
pdbbind_tasks, pdbbind_datasets, transformers = load_pdbbind_pockets(
    split=split, subset=subset)
train_dataset, valid_dataset, test_dataset = pdbbind_datasets