cogdl.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)

cogdl.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 cogdl.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)

The name of the files to find in the self.raw_dir folder in order to skip the download.

property processed_file_names(self)

The name of the files to find in the self.processed_dir folder in order to skip the processing.

read_gtn_data(self, folder)

get(self, idx)

Gets the data object at index idx.

apply_to_device(self, device)

download(self)

Downloads the dataset to the self.raw_dir folder.

process(self)

Processes the dataset to the self.processed_dir folder.

__repr__(self)

class cogdl.datasets.gtn_data.ACM_GTNDataset

Bases: cogdl.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 cogdl.datasets.gtn_data.DBLP_GTNDataset

Bases: cogdl.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 cogdl.datasets.gtn_data.IMDB_GTNDataset

Bases: cogdl.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").