topobench.data.loaders.pointcloud package#

Submodules#

topobench.data.loaders.pointcloud.geometric_shapes module#

Loaders for GeometricShapes datasets.

class topobench.data.loaders.pointcloud.geometric_shapes.GeometricShapesDatasetLoader(parameters: DictConfig)[source]#

Bases: AbstractLoader

Load GeometricShapes dataset.

Parameters:
parametersDictConfig
Configuration parameters containing:

  • data_dir: Root directory for data

load_dataset() → Dataset[source]#

Load GeometricShapes dataset.

Returns:
Dataset

The loaded GeometricShapes dataset.

Raises:
RuntimeError

If dataset loading fails.

topobench.data.loaders.pointcloud.geometric_shapes.rename_pos_to_x(data)[source]#

Rename the ‘pos’ attribute to ‘x’ in a PyG Data object.

This function is needed as a pre_transform for the GeometricShapes dataset so that the ‘pos’ attribute is renamed to ‘x’ properly.

Parameters:
datatorch_geometric.data.Data

The input data.

Returns:
torch_geometric.data.Data

The data with the ‘pos’ attribute renamed to ‘x’.

Module contents#

Init file for PointCloud load module with automated loader discovery.

class topobench.data.loaders.pointcloud.GeometricShapesDatasetLoader(parameters: DictConfig)#

Bases: AbstractLoader

Load GeometricShapes dataset.

Parameters:
parametersDictConfig
Configuration parameters containing:

  • data_dir: Root directory for data

load_dataset() → Dataset#

Load GeometricShapes dataset.

Returns:
Dataset

The loaded GeometricShapes dataset.

Raises:
RuntimeError

If dataset loading fails.