mindspore.dataset.dataloader.Dataset

查看源文件
class mindspore.dataset.dataloader.Dataset[源代码]

所有数据集的基类。

Map style 数据集应该继承自此类。

Map style 数据集表示了一种从键值到数据样本的映射。 子类必须重写 __getitem__() 方法,定义如何根据键值检索样本。 子类可以选择性地重写 __len__() 方法,返回数据集的样本总数。 如果未实现,某些内置采样器和 DataLoader 方法将会不可用。

样例:

>>> from mindspore.dataset.dataloader import Dataset
>>>
>>> class MapStyleDataset(Dataset):
...     def __init__(self, data):
...         self.data = data
...
...     def __getitem__(self, index):
...         return self.data[index]
...
...     def __len__(self):
...         return len(self.data)