Modify DataSetSplitter for only splitting testing/learning