edu.stanford.nlp.coref.neural

Class CategoricalFeatureExtractor

java.lang.Object

edu.stanford.nlp.coref.neural.CategoricalFeatureExtractor

public class CategoricalFeatureExtractor
extends java.lang.Object

Extracts string matching, speaker, distance, and document genre features from mentions.

Author:

Kevin Clark

Constructor Summary

Constructors

Constructor and Description
CategoricalFeatureExtractor(java.util.Properties props, Dictionaries dictionaries)

Method Summary

Modifier and Type	Method and Description
static org.ejml.simple.SimpleMatrix	encodeDistance(int d)
org.ejml.simple.SimpleMatrix	getAnaphoricityFeatures(Mention m, Document document, java.util.Map<java.lang.Integer,java.util.List<Mention>> mentionsByHeadIndex)
org.ejml.simple.SimpleMatrix	getPairFeatures(Pair<java.lang.Integer,java.lang.Integer> pair, Document document, java.util.Map<java.lang.Integer,java.util.List<Mention>> mentionsByHeadIndex)
static java.util.List<java.lang.Integer>	pairwiseFeatures(Document document, Mention m1, Mention m2, Dictionaries dictionaries)

Methods inherited from class java.lang.Object

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

Constructor Detail

CategoricalFeatureExtractor

public CategoricalFeatureExtractor(java.util.Properties props,
                                   Dictionaries dictionaries)

Method Detail

getPairFeatures

public org.ejml.simple.SimpleMatrix getPairFeatures(Pair<java.lang.Integer,java.lang.Integer> pair,
                                                    Document document,
                                                    java.util.Map<java.lang.Integer,java.util.List<Mention>> mentionsByHeadIndex)

pairwiseFeatures

public static java.util.List<java.lang.Integer> pairwiseFeatures(Document document,
                                                                 Mention m1,
                                                                 Mention m2,
                                                                 Dictionaries dictionaries)

getAnaphoricityFeatures

public org.ejml.simple.SimpleMatrix getAnaphoricityFeatures(Mention m,
                                                            Document document,
                                                            java.util.Map<java.lang.Integer,java.util.List<Mention>> mentionsByHeadIndex)

encodeDistance

public static org.ejml.simple.SimpleMatrix encodeDistance(int d)