AmazonCoBuyComputerDataset(raw_dir=None, force_reload=False, verbose=False, transform=None)[source]¶
‘Computer’ part of the AmazonCoBuy dataset for node classification task.
Amazon Computers and Amazon Photo are segments of the Amazon co-purchase graph [McAuley et al., 2015], where nodes represent goods, edges indicate that two goods are frequently bought together, node features are bag-of-words encoded product reviews, and class labels are given by the product category.
Edges: 491,722 (note that the original dataset has 245,778 edges but DGL adds the reverse edges and remove the duplicates, hence with a different number)
Number of classes: 10
Node feature size: 767
raw_dir (str) – Raw file directory to download/contains the input data directory. Default: ~/.dgl/
force_reload (bool) – Whether to reload the dataset. Default: False
verbose (bool) – Whether to print out progress information. Default: True.
transform (callable, optional) – A transform that takes in a
DGLGraphobject and returns a transformed version. The
DGLGraphobject will be transformed before every access.
>>> data = AmazonCoBuyComputerDataset() >>> g = data >>> num_class = data.num_classes >>> feat = g.ndata['feat'] # get node feature >>> label = g.ndata['label'] # get node labels
Get graph by index
idx (int) – Item index
The graph contains:
ndata['feat']: node features
ndata['label']: node labels
- Return type
Number of graphs in the dataset