public SamplingDataSetIterator(DataSet sampleFrom, int batchSize, int totalNumberSamples) {
        super();
        this.sampleFrom = sampleFrom;
        this.batchSize = batchSize;
        this.totalNumberSamples = totalNumberSamples;