docs: private Map<Integer, Double> predict(final double[] x) {
          Map<Integer, Double> result = new HashMap<>();
          for (int i = 0; i < model.weights.length; i++) {
              double y = VectorUtils.dotProduct(x, model.weights[i]);
              y += model.bias[i];
              result.put(i, y);
          }
  
          return result;
      }