datasets.gatne

Module Contents

Classes

GatneDataset	The network datasets “Amazon”, “Twitter” and “YouTube” from the
AmazonDataset	The network datasets “Amazon”, “Twitter” and “YouTube” from the
TwitterDataset	The network datasets “Amazon”, “Twitter” and “YouTube” from the
YouTubeDataset	The network datasets “Amazon”, “Twitter” and “YouTube” from the

Functions

read_gatne_data(folder)

datasets.gatne.read_gatne_data(folder)

class datasets.gatne.GatneDataset(root, name)

Bases: cogdl.data.Dataset

The network datasets “Amazon”, “Twitter” and “YouTube” from the “Representation Learning for Attributed Multiplex Heterogeneous Network” paper.

Args:

root (string): Root directory where the dataset should be saved. name (string): The name of the dataset ("Amazon",

"Twitter", "YouTube").

url = https://github.com/THUDM/GATNE/raw/master/data

property raw_file_names(self)

property processed_file_names(self)

get(self, idx)

download(self)

process(self)

__repr__(self)

class datasets.gatne.AmazonDataset

Bases: datasets.gatne.GatneDataset

The network datasets “Amazon”, “Twitter” and “YouTube” from the “Representation Learning for Attributed Multiplex Heterogeneous Network” paper.

Args:

root (string): Root directory where the dataset should be saved. name (string): The name of the dataset ("Amazon",

"Twitter", "YouTube").

class datasets.gatne.TwitterDataset

Bases: datasets.gatne.GatneDataset

The network datasets “Amazon”, “Twitter” and “YouTube” from the “Representation Learning for Attributed Multiplex Heterogeneous Network” paper.

Args:

root (string): Root directory where the dataset should be saved. name (string): The name of the dataset ("Amazon",

"Twitter", "YouTube").

class datasets.gatne.YouTubeDataset

Bases: datasets.gatne.GatneDataset

The network datasets “Amazon”, “Twitter” and “YouTube” from the “Representation Learning for Attributed Multiplex Heterogeneous Network” paper.

Args:

root (string): Root directory where the dataset should be saved. name (string): The name of the dataset ("Amazon",

"Twitter", "YouTube").