Unverified Commit 21e90cbf authored by Xiangkun Hu's avatar Xiangkun Hu Committed by GitHub
Browse files

[Doc] Fix number of classes in Amazon Cobuy datasets (#2488)

* PPIDataset

* Revert "PPIDataset"

This reverts commit 264bd0c960cfa698a7bb946dad132bf52c2d0c8a.

* Update gnn_benckmark.py

* Hide implementations that may cause confusions to the user

* fix number of classes in Amazon Cobuy datasets
parent 63327acf
...@@ -348,7 +348,7 @@ class AmazonCoBuyComputerDataset(GNNBenchmarkDataset): ...@@ -348,7 +348,7 @@ class AmazonCoBuyComputerDataset(GNNBenchmarkDataset):
- Nodes: 13,752 - Nodes: 13,752
- Edges: 574,418 - Edges: 574,418
- Number of classes: 5 - Number of classes: 10
- Node feature size: 767 - Node feature size: 767
Parameters Parameters
...@@ -390,7 +390,7 @@ class AmazonCoBuyComputerDataset(GNNBenchmarkDataset): ...@@ -390,7 +390,7 @@ class AmazonCoBuyComputerDataset(GNNBenchmarkDataset):
------- -------
int int
""" """
return 5 return 10
class AmazonCoBuyPhotoDataset(GNNBenchmarkDataset): class AmazonCoBuyPhotoDataset(GNNBenchmarkDataset):
...@@ -413,7 +413,7 @@ class AmazonCoBuyPhotoDataset(GNNBenchmarkDataset): ...@@ -413,7 +413,7 @@ class AmazonCoBuyPhotoDataset(GNNBenchmarkDataset):
- Nodes: 7,650 - Nodes: 7,650
- Edges: 287,326 - Edges: 287,326
- Number of classes: 5 - Number of classes: 8
- Node feature size: 745 - Node feature size: 745
Parameters Parameters
...@@ -455,7 +455,7 @@ class AmazonCoBuyPhotoDataset(GNNBenchmarkDataset): ...@@ -455,7 +455,7 @@ class AmazonCoBuyPhotoDataset(GNNBenchmarkDataset):
------- -------
int int
""" """
return 5 return 8
class CoraFull(CoraFullDataset): class CoraFull(CoraFullDataset):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment