magnet.io.load_graph_dataset

magnet.io.load_graph_dataset(dataset_path: str) → MeshDataset

Load mesh graph dataset.

Loads the adjacency, centroids, volumes (and physical groups, if present) of a mesh dataset from a .npz file.

This is intended mainly for training GNNs, as the dataset does not contain the full connectivity data of the meshes.

Parameters:

dataset_path (str) – Path of the .npz file containing the datset, or folder path of the same name containing said file.

Returns:

Loaded dataset.

Return type:

MeshDataset

See also

load_mat_dataset

Load mesh graph dataset from MATLAB files.

generate.2D_dataset

Generate dataset of 2D meshes.

generate.dataset_3D

Generate dataset of 3D meshes.