ray.data.Dataset.union

Dataset.union(*other: List[Dataset]) → Dataset

在行之间连接 数据集。

数据集中块的顺序被保留，参数列表中传递的数据集之间的相对顺序也被保留。

小心

联合数据集不是血统可序列化的。因此，它们不能作为Ray Tune中的可调超参数使用。

示例

>>> import ray
>>> ds1 = ray.data.range(2)
>>> ds2 = ray.data.range(3)
>>> ds1.union(ds2).take_all()
[{'id': 0}, {'id': 1}, {'id': 0}, {'id': 1}, {'id': 2}]

参数:

other – 与此数据集结合的数据集列表。这些数据集必须与此数据集具有相同的模式，否则行为未定义。

返回:

一个包含输入数据集行的新数据集。