datasets.han_data
¶
Module Contents¶
Classes¶
The network datasets “ACM”, “DBLP” and “IMDB” from the |
|
The network datasets “ACM”, “DBLP” and “IMDB” from the |
|
The network datasets “ACM”, “DBLP” and “IMDB” from the |
|
The network datasets “ACM”, “DBLP” and “IMDB” from the |
Functions¶
|
Unpacks the given archive file to the same directory, then (by default) |
|
Create mask. |
-
datasets.han_data.
untar
(path, fname, deleteTar=True)[source]¶ Unpacks the given archive file to the same directory, then (by default) deletes the archive file.
-
class
datasets.han_data.
HANDataset
(root, name)[source]¶ Bases:
cogdl.data.Dataset
The network datasets “ACM”, “DBLP” and “IMDB” from the “Heterogeneous Graph Attention Network” paper.
- Args:
root (string): Root directory where the dataset should be saved. name (string): The name of the dataset (
"han-acm"
,"han-dblp"
,"han-imdb"
).
-
class
datasets.han_data.
ACM_HANDataset
[source]¶ Bases:
datasets.han_data.HANDataset
The network datasets “ACM”, “DBLP” and “IMDB” from the “Heterogeneous Graph Attention Network” paper.
- Args:
root (string): Root directory where the dataset should be saved. name (string): The name of the dataset (
"han-acm"
,"han-dblp"
,"han-imdb"
).
-
class
datasets.han_data.
DBLP_HANDataset
[source]¶ Bases:
datasets.han_data.HANDataset
The network datasets “ACM”, “DBLP” and “IMDB” from the “Heterogeneous Graph Attention Network” paper.
- Args:
root (string): Root directory where the dataset should be saved. name (string): The name of the dataset (
"han-acm"
,"han-dblp"
,"han-imdb"
).
-
class
datasets.han_data.
IMDB_HANDataset
[source]¶ Bases:
datasets.han_data.HANDataset
The network datasets “ACM”, “DBLP” and “IMDB” from the “Heterogeneous Graph Attention Network” paper.
- Args:
root (string): Root directory where the dataset should be saved. name (string): The name of the dataset (
"han-acm"
,"han-dblp"
,"han-imdb"
).