de.datexis.encoder.sector.SectorRESTEncoder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of texoo-encoder-api Show documentation

TeXoo REST encoder interface

There is a newer version: 1.3.3

package de.datexis.encoder.sector;

import de.datexis.encoder.RESTAdapter;
import de.datexis.encoder.SimpleRESTEncoder;
import de.datexis.model.Document;
import de.datexis.model.Sentence;
import de.datexis.model.Span;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.IOException;
import java.util.Collection;

public class SectorRESTEncoder extends SimpleRESTEncoder {
  private static final Logger log = LoggerFactory.getLogger(SectorRESTEncoder.class);

  public static SectorRESTEncoder create(String domain, int port) {
    return new SectorRESTEncoder(new SectorRESTAdapter(domain, port));
  }

  public static SectorRESTEncoder create(String domain, int port, String vectorIdentifier) {
    return new SectorRESTEncoder(new SectorRESTAdapter(domain, port), vectorIdentifier);
  }

  public static SectorRESTEncoder create(String domain, int port, long embeddingVectorSize, int connectTimeout, int readTimeout) {
    return new SectorRESTEncoder(new SectorRESTAdapter(domain, port, embeddingVectorSize, connectTimeout, readTimeout));
  }

  public static SectorRESTEncoder create(String domain, int port, long embeddingVectorSize, int connectTimeout, int readTimeout, String vectorIdentifier) {
    return new SectorRESTEncoder(new SectorRESTAdapter(domain, port, embeddingVectorSize, connectTimeout, readTimeout), vectorIdentifier);
  }

  protected SectorRESTEncoder() {
    super("SEC");
  }
  
  public SectorRESTEncoder(RESTAdapter restAdapter) {
    super("SEC", restAdapter, Sentence.class);
  }

  public SectorRESTEncoder(RESTAdapter restAdapter, String vectorIdentifier) {
    super("SEC", restAdapter, vectorIdentifier, Sentence.class);
  }

  @Override
  public INDArray encodeImpl(String word) throws IOException {
    return encodeValue(word);
  }

  @Override
  public INDArray encodeImpl(Span span) throws IOException {
    return encodeValue(span.getText());
  }

  @Override
  public void encodeEachImpl(Sentence input) throws IOException {
    encodeEach(input, Sentence::toTokenizedString);
  }

  @Override
  public void encodeEachImpl(Document input) throws IOException {
    encodeEach1D(input.getSentences(), Sentence::toTokenizedString);
  }

  @Override
  public void encodeEachImpl(Collection docs) throws IOException {
    for (Document document : docs) {
      encodeEachImpl(document);
    }
  }
}