de.datexis.encoder.impl.ExampleEncoder Maven / Gradle / Ivy

Go to download
package de.datexis.encoder.impl;

import de.datexis.encoder.IEncoder;
import de.datexis.model.Span;
import org.nd4j.linalg.api.ndarray.INDArray;
import org.nd4j.linalg.factory.Nd4j;

/**
 * Example implementation of an 128-dim Encoder / 8-class Decoder
 * @author Sebastian Arnold 
 */
public class ExampleEncoder implements IEncoder {

  @Override
  public long getEmbeddingVectorSize() {
    return 128L;
  }

  @Override
  public INDArray encode(String word) {
    return Nd4j.ones(getEmbeddingVectorSize(), 1);
  }

  @Override
  public INDArray encode(Span span) {
    return encode(span.getText());
  }

  @Override
  public INDArray encode(Iterable spans) {
    INDArray avg = Nd4j.create(getEmbeddingVectorSize(), 1);
    INDArray vec;
    int i = 0;
    for(Span s : spans) {
      vec = encode(s.getText());
      if(vec != null) {
        avg.addi(vec);
        i++;
      }
    }
    return avg.divi(i);
  }

  public long getOutputVectorSize() {
    return 8L;
  }

  public INDArray decode(String word) {
    return Nd4j.ones(getOutputVectorSize(), 1);
  }

  public INDArray decode(Span span) {
    return decode(span.getText());
  }

  public INDArray decode(Iterable spans) {
    INDArray avg = Nd4j.create(getOutputVectorSize(), 1);
    INDArray vec;
    int i = 0;
    for(Span s : spans) {
      vec = decode(s.getText());
      if(vec != null) {
        avg.addi(vec);
        i++;
      }
    }
    return avg.divi(i);
  }
  
}