public class NeuralCorefModel
extends java.lang.Object
implements java.io.Serializable
NeuralCorefAlgorithm
Constructor and Description |
---|
NeuralCorefModel(org.ejml.simple.SimpleMatrix antecedentMatrix,
org.ejml.simple.SimpleMatrix anaphorMatrix,
org.ejml.simple.SimpleMatrix pairFeaturesMatrix,
org.ejml.simple.SimpleMatrix pairwiseFirstLayerBias,
java.util.List<org.ejml.simple.SimpleMatrix> anaphoricityModel,
java.util.List<org.ejml.simple.SimpleMatrix> pairwiseModel,
Embedding wordEmbeddings) |
Modifier and Type | Method and Description |
---|---|
org.ejml.simple.SimpleMatrix |
getAnaphorEmbedding(org.ejml.simple.SimpleMatrix mentionEmbedding) |
java.util.List<org.ejml.simple.SimpleMatrix> |
getAnaphoricityModel() |
double |
getAnaphoricityScore(org.ejml.simple.SimpleMatrix mentionEmbedding,
org.ejml.simple.SimpleMatrix anaphoricityFeatures) |
org.ejml.simple.SimpleMatrix |
getAnaphorMatrix() |
org.ejml.simple.SimpleMatrix |
getAntecedentEmbedding(org.ejml.simple.SimpleMatrix mentionEmbedding) |
org.ejml.simple.SimpleMatrix |
getAntecedentMatrix() |
org.ejml.simple.SimpleMatrix |
getPairFeaturesMatrix() |
org.ejml.simple.SimpleMatrix |
getPairwiseFirstLayerBias() |
java.util.List<org.ejml.simple.SimpleMatrix> |
getPairwiseModel() |
double |
getPairwiseScore(org.ejml.simple.SimpleMatrix antecedentEmbedding,
org.ejml.simple.SimpleMatrix anaphorEmbedding,
org.ejml.simple.SimpleMatrix pairFeatures) |
Embedding |
getWordEmbeddings() |
public NeuralCorefModel(org.ejml.simple.SimpleMatrix antecedentMatrix, org.ejml.simple.SimpleMatrix anaphorMatrix, org.ejml.simple.SimpleMatrix pairFeaturesMatrix, org.ejml.simple.SimpleMatrix pairwiseFirstLayerBias, java.util.List<org.ejml.simple.SimpleMatrix> anaphoricityModel, java.util.List<org.ejml.simple.SimpleMatrix> pairwiseModel, Embedding wordEmbeddings)
public double getAnaphoricityScore(org.ejml.simple.SimpleMatrix mentionEmbedding, org.ejml.simple.SimpleMatrix anaphoricityFeatures)
public double getPairwiseScore(org.ejml.simple.SimpleMatrix antecedentEmbedding, org.ejml.simple.SimpleMatrix anaphorEmbedding, org.ejml.simple.SimpleMatrix pairFeatures)
public org.ejml.simple.SimpleMatrix getAnaphorEmbedding(org.ejml.simple.SimpleMatrix mentionEmbedding)
public org.ejml.simple.SimpleMatrix getAntecedentEmbedding(org.ejml.simple.SimpleMatrix mentionEmbedding)
public Embedding getWordEmbeddings()
public org.ejml.simple.SimpleMatrix getAntecedentMatrix()
public org.ejml.simple.SimpleMatrix getAnaphorMatrix()
public org.ejml.simple.SimpleMatrix getPairFeaturesMatrix()
public org.ejml.simple.SimpleMatrix getPairwiseFirstLayerBias()
public java.util.List<org.ejml.simple.SimpleMatrix> getAnaphoricityModel()
public java.util.List<org.ejml.simple.SimpleMatrix> getPairwiseModel()