提交 30543478 编写于 作者: A anzhengqi

add num_epochs api

上级 3a16925f
...@@ -1130,6 +1130,8 @@ class Dataset: ...@@ -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: ...@@ -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.
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册