Overall model training and evaluation pipeline