decorate.train function

Decorates the final model of a "train" object with model type-dependent elements.