ggfm.datasets.Amazon

class ggfm.datasets.Amazon(root: str | None = None, name: str = 'computers', transform: Callable | None = None, pre_transform: Callable | None = None, force_reload: bool = False, train_ratio: float = 0.1, val_ratio: float = 0.15)

Bases: object

The Amazon Computers and Amazon Photo networks from the “Pitfalls of Graph Neural Network Evaluation” paper. Nodes represent goods and edges represent that two goods are frequently bought together. Given product reviews as bag-of-words node features, the task is to map goods to their respective product category.