python训练模型函数参数