public class EmbeddingExtractor
extends java.lang.Object
implements java.io.Serializable
Constructor and Description |
---|
EmbeddingExtractor(boolean conll,
Embedding staticWordEmbeddings,
Embedding tunedWordEmbeddings,
java.lang.String naEmbedding) |
Modifier and Type | Method and Description |
---|---|
org.ejml.simple.SimpleMatrix |
getDocumentEmbedding(Document document) |
org.ejml.simple.SimpleMatrix |
getMentionEmbeddings(Mention m,
org.ejml.simple.SimpleMatrix docEmbedding) |
org.ejml.simple.SimpleMatrix |
getMentionEmbeddingsForFast(Mention m) |
java.lang.String |
getNAEmbedding() |
Embedding |
getStaticWordEmbeddings() |
Embedding |
getTunedWordEmbeddings() |
org.ejml.simple.SimpleMatrix |
getWordEmbedding(java.lang.String word) |
boolean |
isConll() |
public boolean isConll()
public Embedding getStaticWordEmbeddings()
public Embedding getTunedWordEmbeddings()
public java.lang.String getNAEmbedding()
public org.ejml.simple.SimpleMatrix getDocumentEmbedding(Document document)
public org.ejml.simple.SimpleMatrix getMentionEmbeddingsForFast(Mention m)
public org.ejml.simple.SimpleMatrix getMentionEmbeddings(Mention m, org.ejml.simple.SimpleMatrix docEmbedding)
public org.ejml.simple.SimpleMatrix getWordEmbedding(java.lang.String word)