edu.stanford.nlp.coref.fastneural

Class FastNeuralCorefModel

java.lang.Object

edu.stanford.nlp.coref.fastneural.FastNeuralCorefModel

All Implemented Interfaces:

java.io.Serializable

public class FastNeuralCorefModel
extends java.lang.Object
implements java.io.Serializable

Featurizes and scores mention pairs using a neural network.

Author:

kevinclark

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
FastNeuralCorefModel(EmbeddingExtractor embeddingExtractor, java.util.Map<java.lang.String,java.lang.Integer> pairFeatureIds, java.util.Map<java.lang.String,java.lang.Integer> mentionFeatureIds, java.util.List<org.ejml.simple.SimpleMatrix> weights)

Method Summary

Modifier and Type	Method and Description
java.util.List<org.ejml.simple.SimpleMatrix>	getAllWeights()
EmbeddingExtractor	getEmbeddingExtractor()
java.util.Map<java.lang.String,java.lang.Integer>	getMentionFeatureIds()
java.util.Map<java.lang.String,java.lang.Integer>	getPairFeatureIds()
static FastNeuralCorefModel	loadFromTextFiles(java.lang.String path)
static java.util.Map<java.lang.String,java.lang.Integer>	loadMapFromTextFile(java.lang.String filename)
double	score(Mention antecedent, Mention anaphor, Counter<java.lang.String> antecedentFeatures, Counter<java.lang.String> anaphorFeatures, Counter<java.lang.String> pairFeatures, java.util.Map<java.lang.Integer,org.ejml.simple.SimpleMatrix> antecedentCache, java.util.Map<java.lang.Integer,org.ejml.simple.SimpleMatrix> anaphorCache)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

FastNeuralCorefModel

public FastNeuralCorefModel(EmbeddingExtractor embeddingExtractor,
                            java.util.Map<java.lang.String,java.lang.Integer> pairFeatureIds,
                            java.util.Map<java.lang.String,java.lang.Integer> mentionFeatureIds,
                            java.util.List<org.ejml.simple.SimpleMatrix> weights)

Method Detail

getEmbeddingExtractor

public EmbeddingExtractor getEmbeddingExtractor()

getPairFeatureIds

public java.util.Map<java.lang.String,java.lang.Integer> getPairFeatureIds()

getMentionFeatureIds

public java.util.Map<java.lang.String,java.lang.Integer> getMentionFeatureIds()

getAllWeights

public java.util.List<org.ejml.simple.SimpleMatrix> getAllWeights()

score

public double score(Mention antecedent,
                    Mention anaphor,
                    Counter<java.lang.String> antecedentFeatures,
                    Counter<java.lang.String> anaphorFeatures,
                    Counter<java.lang.String> pairFeatures,
                    java.util.Map<java.lang.Integer,org.ejml.simple.SimpleMatrix> antecedentCache,
                    java.util.Map<java.lang.Integer,org.ejml.simple.SimpleMatrix> anaphorCache)

loadFromTextFiles

public static FastNeuralCorefModel loadFromTextFiles(java.lang.String path)

loadMapFromTextFile

public static java.util.Map<java.lang.String,java.lang.Integer> loadMapFromTextFile(java.lang.String filename)