data.dataset
¶
Module Contents¶
Functions¶
|
|
|
-
class
data.dataset.
Dataset
(root, transform=None, pre_transform=None, pre_filter=None)[source]¶ Bases:
torch.utils.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.
-
property
processed_file_names
(self)[source]¶ The name of the files to find in the
self.processed_dir
folder in order to skip the processing.
-
property
processed_paths
(self)[source]¶ The filepaths to find in the
self.processed_dir
folder in order to skip the processing.