实现第一版功能，加入训练集，打印测试数据的预测结果