com.bbn.nlp.edl.EDLWriter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of nlp-core-open Show documentation

There is a newer version: 4.1.2

package com.bbn.nlp.edl;

import com.google.common.collect.Lists;
import com.google.common.io.CharSink;

import java.io.IOException;
import java.util.List;

/**
 * Writes files in the submission format for the TAC KBP Entity Detection and Linking eval. This is
 * focused on Lorelei NER, so it ignores the KB IDs.
 */
public final class EDLWriter {

  private EDLWriter() {
  }

  public static EDLWriter create() {
    return new EDLWriter();
  }

  public void writeEDLMentions(Iterable edlMentions, CharSink sink) throws IOException {
    final List lines = Lists.newArrayList();

    for (final EDLMention edlMention : edlMentions) {
      lines.add(toLine(edlMention));
    }

    sink.writeLines(lines, "\n");
  }

  private String toLine(final EDLMention edlMention) {
    return edlMention.runId() + "\t" + edlMention.mentionId() + "\t" + edlMention.headString()
        + "\t" + edlMention.documentID() + ":" + edlMention.headOffsets().startInclusive().asInt()
        + "-" + edlMention.headOffsets().endInclusive().asInt() + "\t" + "NIL\t"
        + edlMention.entityType() + "\t" + edlMention.mentionType() + "\t"
        + edlMention.confidence();
  }
}