random_split¶
给定子集合 dataset 的长度数组,随机切分出原数据集合的非重复子集合。
参数¶
dataset (Dataset) - 此参数必须是
paddle.io.Dataset
或paddle.io.IterableDataset
的一个子类实例或实现了__len__
的 Python 对象,用于生成样本下标。默认值为 None。lengths (list) - 总和为原数组长度的,子集合长度数组。
generator (Generator,可选) - 指定采样
data_source
的采样器。默认值为 None。
返回¶
list,返回按给定长度数组描述随机分割的原数据集合的非重复子集合。
代码示例¶
import paddle
from paddle.io import random_split
a_list = paddle.io.random_split(range(10), [3, 7])
print(len(a_list))
# 2
for idx, v in enumerate(a_list[0]):
print(idx, v)
# output of the first subset
# 0 1
# 1 3
# 2 9
for idx, v in enumerate(a_list[1]):
print(idx, v)
# output of the second subset
# 0 5
# 1 7
# 2 8
# 3 6
# 4 0
# 5 2
# 6 4