gammagl.datasets.AMiner¶
- class AMiner(root: str | None = None, transform: Callable | None = None, pre_transform: Callable | None = None, pre_filter: Callable | None = None, force_reload: bool = False)[source]¶
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.- Parameters:
root (str, optional) – Root directory where the dataset should be saved.
transform (callable, optional) – A function/transform that takes in an
gammagl.data.HeteroGraph
object and returns a transformed version. The data object will be transformed before every access. (default:None
)pre_transform (callable, optional) – A function/transform that takes in an
gammagl.data.HeteroGraph
object and returns a transformed version. The data object will be transformed before being saved to disk. (default:None
)(bool (force_reload) – (default:
False
)optional) (Whether to re-process the dataset.) – (default:
False
)
- url = 'https://www.dropbox.com/s/1bnz8r7mofx0osf/net_aminer.zip?dl=1'¶
- y_url = 'https://www.dropbox.com/s/nkocx16rpl4ydde/label.zip?dl=1'¶
- property raw_file_names: List[str]¶
The name of the files in the
self.raw_dir
folder that must be present in order to skip downloading.