cogdl.datasets.gcc_data
¶
Module Contents¶
Classes¶
Dataset base class for creating graph datasets. |
|
Dataset base class for creating graph datasets. |
-
class
cogdl.datasets.gcc_data.
Edgelist
(root, name)[source]¶ Bases:
cogdl.data.Dataset
Dataset base class for creating graph datasets. See here for the accompanying tutorial.
- Args:
root (string): Root directory where the dataset should be saved. transform (callable, optional): A function/transform that takes in an
cogdl.data.Data
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
cogdl.data.Data
object and returns a transformed version. The data object will be transformed before being saved to disk. (default:None
)- pre_filter (callable, optional): A function that takes in an
cogdl.data.Data
object and returns a boolean value, indicating whether the data object should be included in the final dataset. (default:None
)
-
property
raw_file_names
(self)[source]¶ The name of the files to find in the
self.raw_dir
folder in order to skip the download.
-
class
cogdl.datasets.gcc_data.
USAAirportDataset
[source]¶ Bases:
cogdl.datasets.gcc_data.Edgelist
Dataset base class for creating graph datasets. See here for the accompanying tutorial.
- Args:
root (string): Root directory where the dataset should be saved. transform (callable, optional): A function/transform that takes in an
cogdl.data.Data
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
cogdl.data.Data
object and returns a transformed version. The data object will be transformed before being saved to disk. (default:None
)- pre_filter (callable, optional): A function that takes in an
cogdl.data.Data
object and returns a boolean value, indicating whether the data object should be included in the final dataset. (default:None
)