修改main函数，提取样本进行训练