models for training

GBM model added