writer = xof.createXMLStreamWriter(os);
int trainingDataCount = extractXMLData(reader, writer, trainingDataSize);
os.close();
os = new FileOutputStream(testOutputFile);
writer = xof.createXMLStreamWriter(os);
int testDataCount = extractXMLData(reader, writer, testDataSize);
os.close();
log.info("Extracted " + trainingDataCount + " rows of training data");
log.info("Extracted " + testDataCount + " rows of test data");