17.8 把数据分为训练集和测试集