code: @Override
      public DataSet next() {
          counter.incrementAndGet();
  
          DataSet ds = new DataSet(baseFeatures, baseLabels);
  
          return ds;
      }
docs: Returns the next element in the iteration.
  
  @return the next element in the iteration