code: public Map<String, Double> convertMapKey(Map<Integer, Double> probs) {
          Map<String, Double> stringKeyProb = new HashMap<>();
          probs.entrySet().forEach(e -> stringKeyProb.put(getLabel(e.getKey()), e.getValue()));
          return stringKeyProb;
      }
docs: Convert Index to actual string.