public class MentionDetectionClassifier extends Object implements Serializable
| Modifier and Type | Field and Description |
|---|---|
RandomForest |
rf |
| Constructor and Description |
|---|
MentionDetectionClassifier(RandomForest rf) |
| Modifier and Type | Method and Description |
|---|---|
void |
classifyMentions(List<List<Mention>> predictedMentions,
Dictionaries dict,
Properties props) |
static Counter<String> |
extractFeatures(Mention p,
Set<Mention> shares,
Set<String> neStrings,
Dictionaries dict,
Properties props) |
static MentionDetectionClassifier |
loadMentionDetectionClassifier(String filename) |
double |
probabilityOf(Mention p,
Set<Mention> shares,
Set<String> neStrings,
Dictionaries dict,
Properties props) |
public RandomForest rf
public MentionDetectionClassifier(RandomForest rf)
public static Counter<String> extractFeatures(Mention p, Set<Mention> shares, Set<String> neStrings, Dictionaries dict, Properties props)
public static MentionDetectionClassifier loadMentionDetectionClassifier(String filename) throws ClassNotFoundException, IOException
ClassNotFoundExceptionIOExceptionpublic double probabilityOf(Mention p, Set<Mention> shares, Set<String> neStrings, Dictionaries dict, Properties props)
public void classifyMentions(List<List<Mention>> predictedMentions, Dictionaries dict, Properties props)