Updated train_model.py to be compatible with phenotype classification

1 job for master in 14 minutes and 57 seconds (queued for 2 seconds)