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)