class Mitie::NER

Attributes

pointer[R]

Public Class Methods

finalize(pointer) click to toggle source
# File lib/mitie/ner.rb, line 34
def self.finalize(pointer)
  # must use proc instead of stabby lambda
  proc { FFI.mitie_free(pointer) }
end
new(path) click to toggle source
# File lib/mitie/ner.rb, line 5
def initialize(path)
  # better error message
  raise ArgumentError, "File does not exist" unless File.exist?(path)
  @pointer = FFI.mitie_load_named_entity_extractor(path)
  ObjectSpace.define_finalizer(self, self.class.finalize(pointer))
end

Public Instance Methods

doc(text) click to toggle source
# File lib/mitie/ner.rb, line 18
def doc(text)
  Document.new(self, text)
end
entities(text) click to toggle source
# File lib/mitie/ner.rb, line 22
def entities(text)
  doc(text).entities
end
tags() click to toggle source
# File lib/mitie/ner.rb, line 12
def tags
  FFI.mitie_get_num_possible_ner_tags(pointer).times.map do |i|
    FFI.mitie_get_named_entity_tagstr(pointer, i).to_s
  end
end
tokens(text) click to toggle source
# File lib/mitie/ner.rb, line 26
def tokens(text)
  doc(text).tokens
end
tokens_with_offset(text) click to toggle source
# File lib/mitie/ner.rb, line 30
def tokens_with_offset(text)
  doc(text).tokens_with_offset
end