merge
training and prediction