Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!5179 add create_dict_iterator parameter num_epochs api 

Merge pull request !5179 from anzhengqi/add-num_epochs-api-r0.7
This commit is contained in:
mindspore-ci-bot 2020-08-26 10:24:08 +08:00 committed by Gitee
parent 0bf7f1f39c 3c4feaa4f4
commit 185b25c6de
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
mindspore/dataset/engine/datasets.py
View File

@ -1130,6 +1130,8 @@ class Dataset:
Args:
columns (list[str], optional): List of columns to be used to specify the order of columns
(default=None, means all columns).
num_epochs (int, optional): max epochs that iterator can be iteratered,
if num_epochs = -1, iterator can be iteratered infinit epochs (default=-1)
Returns:
Iterator, list of ndarray.
@ -1155,6 +1157,10 @@ class Dataset:
The data retrieved will be a dictionary. The order
of the columns in the dictionary may not be the same as the original order.
Args:
num_epochs (int, optional): max epochs that iterator can be iteratered,
if num_epochs = -1, iterator can be iteratered infinit epochs (default=-1)
Returns:
Iterator, dictionary of column_name-ndarray pair.