ggfm.datasets.AMiner¶

class ggfm.datasets.AMiner(root: str | None = None, transform: Callable | None = None, pre_transform: Callable | None = None, pre_filter: Callable | None = None, force_reload: bool = False)[source]¶

Bases: object

The heterogeneous AMiner dataset from the “metapath2vec: Scalable Representation Learning for Heterogeneous Networks” paper, consisting of nodes from type "paper", "author" and "venue". Aminer is a heterogeneous graph containing three types of entities - author (1,693,531 nodes), venue (3,883 nodes), and paper (3,194,405 nodes). Venue categories and author research interests are available as ground truth labels for a subset of nodes.