docs: @Override
      public DataSet vectorize(String text, String label) {
          INDArray input = transform(text);
          INDArray labelMatrix = FeatureUtil.toOutcomeVector(labelsSource.indexOf(label), labelsSource.size());
  
          return new DataSet(input, labelMatrix);
      }