想请问如何取出数据集的一部分进行训练？（小白）

[DipperZ]

您好，我想在本地计算机上跑一些基础的实验，但是criteo和avazu数据集太过庞大，本地设备无法支持，想请问一下有什么方法能取出数据集的一部分进行训练呢，例如取出criteo数据的前10k行或者5%之类的，我在配置文件中将split_ratio设置成了[0.0008,0.0001,0.0002],但还是跑不动