标签： python split dataset pytorch
我有一个pytorch数据集，每个类具有不同数量的元素。 我如何获得具有由给定数量的元素组成的类的两个子集（这些元素是随机获取的）。元素数是原始数据集中类的元素数的固定百分比（每个类固定）吗？
举个例子：
原始数据集（class1 = 9个元素，class2 = 15个元素）
固定百分比= 1/3
dataset1（class1 = 3个元素，class2 = 5个元素），
数据集2（class1 = 6个元素class2 = 10个元素）