Separation of train and predict processes.