train_models function

Train Individual Models