mlx_graphs.datasets.SuperPixelDataset

mlx_graphs.datasets.SuperPixelDataset#

class mlx_graphs.datasets.SuperPixelDataset(name: Literal['MNIST', 'CIFAR10'], split: Literal['train', 'test'], use_features: bool = False, base_dir: str | None = None)[source]#

Bases: Dataset

MNIST and CIFAR10 superpixel datasets for graph classification tasks converted fromt the original MINST and CIFAR10 images.

The datasets were introduced in http://arxiv.org/abs/2003.00982.

Parameters:

  • name (Literal['MNIST', 'CIFAR10']) – name of the selected dataset

  • split (Literal['train', 'test']) – split of the dataset to load

  • use_features (bool) – if True, the adjacency matrix is computed from superpixels locations and features. If False, only from superpixels locations. Defaults to False.

  • base_dir (Optional[str]) – directory where to store the datasets