重构训练模型