gordicaleksa
diff --git a/‎The Annotated GAT (Cora).ipynb
Lines changed: 23 additions & 24 deletions b/‎The Annotated GAT (Cora).ipynb
Lines changed: 23 additions & 24 deletions
diff --git a/‎The Annotated GAT (PPI).ipynb
Lines changed: 39186 additions & 109535 deletions b/‎The Annotated GAT (PPI).ipynb
Lines changed: 39186 additions & 109535 deletions
diff --git a/‎playground.py
Lines changed: 2 additions & 1 deletion b/‎playground.py
Lines changed: 2 additions & 1 deletion
@@ -155,7 +155,8 @@ def visualize_graph_dataset(dataset_name):
     config = {
         'dataset_name': dataset_name,  # Cora or PPI
         'layer_type': LayerType.IMP3,  # don't care, but it's needed for load_graph_data function to work
-        'should_visualize': True  # visualize the dataset
+        'should_visualize': True,  # visualize the dataset
+        'ppi_load_test_only': True  # only used for PPI, let's just load the test graphs
     }
     load_graph_data(config, device)
Original file line number	Diff line number	Diff line change
`@@ -155,7 +155,8 @@ def visualize_graph_dataset(dataset_name):`
`155`	`155`	`config = {`
`156`	`156`	`'dataset_name': dataset_name, # Cora or PPI`
`157`	`157`	`'layer_type': LayerType.IMP3, # don't care, but it's needed for load_graph_data function to work`
`158`		`- 'should_visualize': True # visualize the dataset`
	`158`	`+ 'should_visualize': True, # visualize the dataset`
	`159`	`+ 'ppi_load_test_only': True # only used for PPI, let's just load the test graphs`
`159`	`160`	`}`
`160`	`161`	`load_graph_data(config, device)`
`161`	`162`