[TrainModel] Fixed message constructors