Added models for trainingByModel