public class Embedding
extends java.lang.Object
implements java.io.Serializable
<lmthang@stanford.edu>
, John Bauer, Richard Socher, Kevin ClarkConstructor and Description |
---|
Embedding(java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> wordVectors) |
Embedding(java.lang.String wordVectorFile) |
Embedding(java.lang.String wordVectorFile,
int embeddingSize) |
Embedding(java.lang.String wordFile,
java.lang.String vectorFile) |
Embedding(java.lang.String wordFile,
java.lang.String vectorFile,
int embeddingSize) |
Modifier and Type | Method and Description |
---|---|
boolean |
containsWord(java.lang.String word) |
java.util.Set<java.util.Map.Entry<java.lang.String,org.ejml.simple.SimpleMatrix>> |
entrySet() |
org.ejml.simple.SimpleMatrix |
get(java.lang.String word) |
int |
getEmbeddingSize() |
org.ejml.simple.SimpleMatrix |
getEndWordVector() |
org.ejml.simple.SimpleMatrix |
getStartWordVector() |
org.ejml.simple.SimpleMatrix |
getUnknownWordVector() |
java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> |
getWordVectors() |
java.util.Set<java.lang.String> |
keySet() |
void |
setWordVectors(java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> wordVectors) |
int |
size() |
java.util.Collection<org.ejml.simple.SimpleMatrix> |
values() |
void |
writeToFile(java.lang.String filename) |
public Embedding(java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> wordVectors)
public Embedding(java.lang.String wordVectorFile)
public Embedding(java.lang.String wordVectorFile, int embeddingSize)
public Embedding(java.lang.String wordFile, java.lang.String vectorFile)
public Embedding(java.lang.String wordFile, java.lang.String vectorFile, int embeddingSize)
public void writeToFile(java.lang.String filename) throws java.io.IOException
java.io.IOException
public int size()
public java.util.Collection<org.ejml.simple.SimpleMatrix> values()
public java.util.Set<java.lang.String> keySet()
public java.util.Set<java.util.Map.Entry<java.lang.String,org.ejml.simple.SimpleMatrix>> entrySet()
public org.ejml.simple.SimpleMatrix get(java.lang.String word)
public boolean containsWord(java.lang.String word)
public org.ejml.simple.SimpleMatrix getStartWordVector()
public org.ejml.simple.SimpleMatrix getEndWordVector()
public org.ejml.simple.SimpleMatrix getUnknownWordVector()
public java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> getWordVectors()
public int getEmbeddingSize()
public void setWordVectors(java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> wordVectors)