将数据集转换为训练的格式