EXP-124. Add the train phase to model definitions.

issue: https://jira.kiji.org/browse/EXP-124
review: https://review.kiji.org/r/887

Reviewed by: kiyan