I don't think there is anything wrong with expecting two file paths and essentially wrapping a version of ImageDataSet, TextDataSet and PartitionedDataSet all together!