Пример #1
0
  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++;
    }
  }
Пример #2
0
  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);
  }