Package edu.stanford.nlp.coref.fastneural

Class FastNeuralCorefModel

java.lang.Object

edu.stanford.nlp.coref.fastneural.FastNeuralCorefModel

All Implemented Interfaces:

Serializable

public class FastNeuralCorefModel
extends Object
implements Serializable

Featurizes and scores mention pairs using a neural network.

Author:

kevinclark

See Also:

Serialized Form

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method	Description
List<org.ejml.simple.SimpleMatrix>	getAllWeights()
EmbeddingExtractor	getEmbeddingExtractor()
Map<String,Integer>	getMentionFeatureIds()
Map<String,Integer>	getPairFeatureIds()
static FastNeuralCorefModel	loadFromTextFiles(String path)
static Map<String,Integer>	loadMapFromTextFile(String filename)
double	score(Mention antecedent, Mention anaphor, Counter<String> antecedentFeatures, Counter<String> anaphorFeatures, Counter<String> pairFeatures, Map<Integer,org.ejml.simple.SimpleMatrix> antecedentCache, Map<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,
                            Map<String,Integer> pairFeatureIds,
                            Map<String,Integer> mentionFeatureIds,
                            List<org.ejml.simple.SimpleMatrix> weights)

Method Detail

getEmbeddingExtractor

public EmbeddingExtractor getEmbeddingExtractor()

getPairFeatureIds

public Map<String,Integer> getPairFeatureIds()

getMentionFeatureIds

public Map<String,Integer> getMentionFeatureIds()

getAllWeights

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

score

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

loadFromTextFiles

public static FastNeuralCorefModel loadFromTextFiles(String path)

loadMapFromTextFile

public static Map<String,Integer> loadMapFromTextFile(String filename)