EXP-125. Add the train phase to model environments.

Changes prepare phase to match train.

issue: https://jira.kiji.org/browse/EXP-125
review: https://review.kiji.org/r/904