mindspore.dataset.Dataset.reset

查看源文件
mindspore.dataset.Dataset.reset()[源代码]

重置下一个epoch的数据集对象。

样例:

>>> import mindspore.dataset as ds
>>> mind_dataset_dir = ["/path/to/mind_dataset_file"]
>>> dataset = ds.MindDataset(dataset_files=mind_dataset_dir)
>>> for _ in range(5):
...     num_iter = 0
...     for data in dataset.create_tuple_iterator(num_epochs=1, output_numpy=True):
...         num_iter += 1
...     dataset.reset()