docs: @Override
      public List<String> predict(DataSet dataSet) {
          int[] intRet = predict(dataSet.getFeatures());
          List<String> ret = new ArrayList<>();
          for (int i : intRet) {
              ret.add(i, dataSet.getLabelName(i));
          }
          return ret;
      }