ray.data.Dataset.take_all

Dataset.take_all(limit: int | None = None) → List[Dict[str, Any]]

返回此 数据集 中的所有行。

此方法适用于检查小型数据集。

警告

take_all() 将整个数据集移动到调用者的机器上。如果数据集很大，此方法可能会导致调用者出现 OutOfMemory 错误。

备注

此操作将触发对此数据集执行的延迟转换。

示例

>>> import ray
>>> ds = ray.data.range(5)
>>> ds.take_all()
[{'id': 0}, {'id': 1}, {'id': 2}, {'id': 3}, {'id': 4}]

时间复杂度：O(数据集大小)

参数:

limit – 如果大小超过指定限制，则引发错误。

返回:

数据集中所有行的列表。

参见

take()

调用此方法以返回特定数量的行。