datasets.gtn_data

Module Contents

Classes

GTNDataset	The network datasets “ACM”, “DBLP” and “IMDB” from the
ACM_GTNDataset	The network datasets “ACM”, “DBLP” and “IMDB” from the
DBLP_GTNDataset	The network datasets “ACM”, “DBLP” and “IMDB” from the
IMDB_GTNDataset	The network datasets “ACM”, “DBLP” and “IMDB” from the

Functions

untar(path, fname, deleteTar=True)

Unpacks the given archive file to the same directory, then (by default)

datasets.gtn_data.untar(path, fname, deleteTar=True)

Unpacks the given archive file to the same directory, then (by default) deletes the archive file.

class datasets.gtn_data.GTNDataset(root, name)

Bases: cogdl.data.Dataset

The network datasets “ACM”, “DBLP” and “IMDB” from the “Graph Transformer Networks” paper.

Args:

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

"gtn-dblp", "gtn-imdb").

property raw_file_names(self)

property processed_file_names(self)

read_gtn_data(self, folder)

get(self, idx)

apply_to_device(self, device)

download(self)

process(self)

__repr__(self)

class datasets.gtn_data.ACM_GTNDataset

Bases: datasets.gtn_data.GTNDataset

The network datasets “ACM”, “DBLP” and “IMDB” from the “Graph Transformer Networks” paper.

Args:

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

"gtn-dblp", "gtn-imdb").

class datasets.gtn_data.DBLP_GTNDataset

Bases: datasets.gtn_data.GTNDataset

The network datasets “ACM”, “DBLP” and “IMDB” from the “Graph Transformer Networks” paper.

Args:

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

"gtn-dblp", "gtn-imdb").

class datasets.gtn_data.IMDB_GTNDataset

Bases: datasets.gtn_data.GTNDataset

The network datasets “ACM”, “DBLP” and “IMDB” from the “Graph Transformer Networks” paper.

Args:

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

"gtn-dblp", "gtn-imdb").