Implement mini-batch SGD and train/test data splitting.