Removed the data splitting from the training script.