code: @Override
      public List<String> predict(org.nd4j.linalg.dataset.api.DataSet dataSet) {
          Preconditions.checkState(dataSet.getLabelNamesList() != null, "This method can only be used when the DataSet contains a label name list");
          int[] intRet = predict(dataSet.getFeatures());
          List<String> ret = new ArrayList<>();
          for (int i = 0; i < intRet.length; i++) {
              ret.add(i, dataSet.getLabelName(intRet[i]));
          }
          return ret;
      }
docs: As per {@link #predict(INDArray)} but the returned values are looked up from the list of label names
  in the provided DataSet