Updated train_model.py to be compatible with phenotype classification

2 jobs for master in 19 minutes and 35 seconds (queued for 1 second)