public class DVModel
extends java.lang.Object
implements java.io.Serializable
Modifier and Type | Field and Description |
---|---|
TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> |
binaryScore |
TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> |
binaryTransform |
java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> |
unaryScore |
java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> |
unaryTransform |
java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> |
wordVectors |
Constructor and Description |
---|
DVModel(Options op,
Index<java.lang.String> stateIndex,
UnaryGrammar unaryGrammar,
BinaryGrammar binaryGrammar) |
DVModel(TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> binaryTransform,
java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> unaryTransform,
TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> binaryScore,
java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> unaryScore,
java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> wordVectors,
Options op) |
Modifier and Type | Method and Description |
---|---|
void |
addRandomBinaryMatrix(java.lang.String leftBasic,
java.lang.String rightBasic) |
void |
addRandomUnaryMatrix(java.lang.String childBasic) |
java.lang.String |
basicCategory(java.lang.String category) |
int |
binaryScoreIndex(java.lang.String leftChild,
java.lang.String rightChild) |
int |
binaryTransformIndex(java.lang.String leftChild,
java.lang.String rightChild) |
void |
filterRulesForBatch(java.util.Collection<Tree> trees)
Filters the transform and score rules so that we only have the
ones which appear in the trees given
|
void |
filterRulesForBatch(java.util.Map<Tree,byte[]> compressedTrees) |
void |
filterRulesForBatch(TwoDimensionalSet<java.lang.String,java.lang.String> binaryRules,
java.util.Set<java.lang.String> unaryRules,
java.util.Set<java.lang.String> words) |
org.ejml.simple.SimpleMatrix |
getEndWordVector() |
org.ejml.simple.SimpleMatrix |
getScoreWForNode(Tree node) |
org.ejml.simple.SimpleMatrix |
getStartWordVector() |
org.ejml.simple.SimpleMatrix |
getUnknownWordVector() |
java.lang.String |
getVocabWord(java.lang.String word) |
org.ejml.simple.SimpleMatrix |
getWForNode(Tree node) |
org.ejml.simple.SimpleMatrix |
getWordVector(java.lang.String word) |
Pair<java.lang.String,java.lang.String> |
indexToBinaryScore(int pos) |
Pair<java.lang.String,java.lang.String> |
indexToBinaryTransform(int pos) |
java.lang.String |
indexToUnaryScore(int pos) |
java.lang.String |
indexToUnaryTransform(int pos) |
double[] |
paramsToVector() |
double[] |
paramsToVector(double scale) |
void |
printAllMatrices(java.io.PrintStream out) |
void |
printMatrixNames(java.io.PrintStream out) |
void |
printMatrixStats(java.io.PrintStream out) |
void |
printParameterType(int pos,
java.io.PrintStream out)
Prints to stdout the type and key for the given location in the parameter stack
|
void |
readWordVectors() |
void |
setRulesForTrainingSet(java.util.List<Tree> sentences,
java.util.Map<Tree,byte[]> compressedTrees) |
int |
totalParamSize() |
int |
unaryScoreIndex(java.lang.String child) |
int |
unaryTransformIndex(java.lang.String child) |
void |
vectorToParams(double[] theta) |
public TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> binaryTransform
public java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> unaryTransform
public TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> binaryScore
public java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> unaryScore
public java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> wordVectors
public DVModel(Options op, Index<java.lang.String> stateIndex, UnaryGrammar unaryGrammar, BinaryGrammar binaryGrammar)
op
- the parameters of the parserpublic DVModel(TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> binaryTransform, java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> unaryTransform, TwoDimensionalMap<java.lang.String,java.lang.String,org.ejml.simple.SimpleMatrix> binaryScore, java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> unaryScore, java.util.Map<java.lang.String,org.ejml.simple.SimpleMatrix> wordVectors, Options op)
public void addRandomUnaryMatrix(java.lang.String childBasic)
public void addRandomBinaryMatrix(java.lang.String leftBasic, java.lang.String rightBasic)
public void setRulesForTrainingSet(java.util.List<Tree> sentences, java.util.Map<Tree,byte[]> compressedTrees)
public void filterRulesForBatch(java.util.Collection<Tree> trees)
public void filterRulesForBatch(java.util.Map<Tree,byte[]> compressedTrees)
public void filterRulesForBatch(TwoDimensionalSet<java.lang.String,java.lang.String> binaryRules, java.util.Set<java.lang.String> unaryRules, java.util.Set<java.lang.String> words)
public java.lang.String basicCategory(java.lang.String category)
public void readWordVectors()
public int totalParamSize()
public double[] paramsToVector(double scale)
public double[] paramsToVector()
public void vectorToParams(double[] theta)
public org.ejml.simple.SimpleMatrix getWForNode(Tree node)
public org.ejml.simple.SimpleMatrix getScoreWForNode(Tree node)
public org.ejml.simple.SimpleMatrix getStartWordVector()
public org.ejml.simple.SimpleMatrix getEndWordVector()
public org.ejml.simple.SimpleMatrix getWordVector(java.lang.String word)
public java.lang.String getVocabWord(java.lang.String word)
public org.ejml.simple.SimpleMatrix getUnknownWordVector()
public void printMatrixNames(java.io.PrintStream out)
public void printMatrixStats(java.io.PrintStream out)
public void printAllMatrices(java.io.PrintStream out)
public int binaryTransformIndex(java.lang.String leftChild, java.lang.String rightChild)
public int unaryTransformIndex(java.lang.String child)
public int binaryScoreIndex(java.lang.String leftChild, java.lang.String rightChild)
public int unaryScoreIndex(java.lang.String child)
public Pair<java.lang.String,java.lang.String> indexToBinaryTransform(int pos)
public java.lang.String indexToUnaryTransform(int pos)
public Pair<java.lang.String,java.lang.String> indexToBinaryScore(int pos)
public java.lang.String indexToUnaryScore(int pos)
public void printParameterType(int pos, java.io.PrintStream out)