docs: public double similarityToLabel(List<VocabWord> document, String label) {
          if (document.isEmpty())
              throw new IllegalStateException("Document has no words inside");
  
          /*
          INDArray arr = Nd4j.create(document.size(), this.layerSize);
          for (int i = 0; i < document.size(); i++) {
              arr.putRow(i, getWordVectorMatrix(document.get(i).getWord()));
          }*/
  
          INDArray docMean = inferVector(document); //arr.mean(0);
  
          INDArray otherVec = getWordVectorMatrix(label);
          double sim = Transforms.cosineSim(docMean, otherVec);
          return sim;
      }