5.2.2 分割训练集和测试集