OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

!5125 add create_dict_iterator parameter num_epochs api 

Merge pull request !5125 from anzhengqi/add-num_epochs-api
This commit is contained in:
mindspore-ci-bot 2020-08-25 20:31:35 +08:00 committed by Gitee
parent 167dc5f09e 305434786a
commit c7c5835800
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: Args:
columns (list[str], optional): List of columns to be used to specify the order of columns columns (list[str], optional): List of columns to be used to specify the order of columns
(default=None, means all 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: Returns:
Iterator, list of ndarray. Iterator, list of ndarray.
@ -1155,6 +1157,10 @@ class Dataset:
The data retrieved will be a dictionary. The order The data retrieved will be a dictionary. The order
of the columns in the dictionary may not be the same as the original 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: Returns:
Iterator, dictionary of column_name-ndarray pair. Iterator, dictionary of column_name-ndarray pair.