private static void applyTemplatesToState(
List<AbstractTemplate<State>> templates, State state, boolean force) {
log.debug("All entities: %s", state.getEntities());
log.debug("All entityIDs: %s", state.getEntityIDs());
log.debug("Non-fixed entities: %s", state.getEditableEntities());
log.debug("Non-fixed entityIDs: %s", state.getEditableEntityIDs());
for (AbstractTemplate<State> t : templates) {
t.applyTo(state, force);
}
Collection<AbstractFactor> factors = state.getFactorGraph().getFactors();
int i = 0;
for (AbstractFactor factor : factors) {
log.debug("\tFactor %s", factor);
Vector v = factor.getFeatureVector();
for (String f : v.getFeatureNames()) {
log.debug(
"\t%s:\t%s", EvaluationUtil.featureWeightFormat.format(v.getValueOfFeature(f)), f);
}
i++;
}
}
public static void main(String[] args) {
Corpus<? extends LabeledDocument<State, State>> corpus = null;
switch (1) {
case 0:
corpus = DummyData.getDummyData();
break;
case 1:
corpus = BioNLPLoader.loadBioNLP2013Train(false);
break;
default:
break;
}
LabeledDocument<State, State> doc = null;
for (LabeledDocument<State, State> tmpDoc : corpus.getDocuments()) {
if (tmpDoc.getGoldResult().getEntities().size() > 0) {
doc = tmpDoc;
break;
}
}
log.debug("Content: %s (%s)", doc.getContent(), doc.getContent().length());
log.debug("Tokens: %s", doc.getTokens());
log.debug("State: %s", doc.getGoldResult());
List<AbstractTemplate<State>> templates =
Arrays.asList(new MorphologicalTemplate(), new ContextTemplate(), new RelationTemplate());
State state = new State(doc.getGoldResult());
applyTemplatesToState(templates, state, false);
// state.markAsUnchanged();
log.debug("");
log.debug("########### Modify State ###########");
log.debug("");
EntityAnnotation e = new ArrayList<>(state.getEntities()).get(0);
e.setType(new EntityType("Banana"));
applyTemplatesToState(templates, state, false);
}