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

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