public class KBPStatisticalExtractor extends java.lang.Object implements KBPRelationExtractor, java.io.Serializable
KBPRelationExtractor.Accuracy, KBPRelationExtractor.KBPInput, KBPRelationExtractor.NERTag, KBPRelationExtractor.RelationType
Modifier and Type | Field and Description |
---|---|
Classifier<java.lang.String,java.lang.String> |
classifier
The implementing classifier of this extractor.
|
static java.lang.String |
MODEL_FILE |
static java.util.Optional<java.lang.String> |
PREDICTIONS |
static java.io.File |
TEST_FILE |
static java.io.File |
TRAIN_FILE |
NO_RELATION
Constructor and Description |
---|
KBPStatisticalExtractor(Classifier<java.lang.String,java.lang.String> classifier)
Create a new KBP relation extractor, from the given implementing classifier.
|
Modifier and Type | Method and Description |
---|---|
Pair<java.lang.String,java.lang.Double> |
classify(KBPRelationExtractor.KBPInput input)
Score the given input, returning both the classification decision and the
probability of that decision.
|
static Counter<java.lang.String> |
features(KBPRelationExtractor.KBPInput input) |
static void |
main(java.lang.String[] args) |
static Classifier<java.lang.String,java.lang.String> |
trainMultinomialClassifier(GeneralDataset<java.lang.String,java.lang.String> dataset,
int featureThreshold,
double sigma)
Train a multinomial classifier off of the provided dataset.
|
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
computeAccuracy, readDataset
@ArgumentParser.Option(name="train", gloss="The dataset to train on") public static java.io.File TRAIN_FILE
@ArgumentParser.Option(name="test", gloss="The dataset to test on") public static java.io.File TEST_FILE
@ArgumentParser.Option(name="model", gloss="The dataset to test on") public static java.lang.String MODEL_FILE
@ArgumentParser.Option(name="predictions", gloss="Dump model predictions to this file") public static java.util.Optional<java.lang.String> PREDICTIONS
public final Classifier<java.lang.String,java.lang.String> classifier
public KBPStatisticalExtractor(Classifier<java.lang.String,java.lang.String> classifier)
classifier
- The implementing classifier.public static Counter<java.lang.String> features(KBPRelationExtractor.KBPInput input)
public static Classifier<java.lang.String,java.lang.String> trainMultinomialClassifier(GeneralDataset<java.lang.String,java.lang.String> dataset, int featureThreshold, double sigma)
dataset
- The dataset to train the classifier off of.public Pair<java.lang.String,java.lang.Double> classify(KBPRelationExtractor.KBPInput input)
classify
in interface KBPRelationExtractor
input
- The input to classify.public static void main(java.lang.String[] args) throws java.io.IOException, java.lang.ClassNotFoundException
java.io.IOException
java.lang.ClassNotFoundException