Treat - Coreference Extraction With Graph Colorization

treat-coref-colorization.rb

    require 'treat'
    Treat.sweeten!
    p = Paragraph "Obama and Sarkozy met on January 1st to investigate the possibility of a new rescue plan." +
        "President Sarkozy is to meet Merkel next Tuesday in Berlin."

    p.do(
      :coreferences,
      :visualize => [
        :dot, {
          :file => 'ner-coref-extraction.dot',
          :colors => {
            :green => lambda do |entity| 
              entity.has?(:named_entity_tag) && 
              entity.named_entity_tag == :person
            end,
            :blue => lambda do |entity| 
              entity.has?(:named_entity_tag) &&
              entity.named_entity_tag == :date
            end,
            :red => lambda do |entity| 
              entity.has?(:named_entity_tag) &&
              entity.named_entity_tag == :location
            end
          }
        }
      ]
    )