KIJIREPO-55. Update the model_container to support specifying training parameters

issue: https://jira.kiji.org/browse/KIJIREPO-55
review: https://review.kiji.org/r/1872