modified train_model
added cat_model_lr.pkl