Method: AnyStyle::Finder#label

Defined in:
lib/anystyle/finder.rb

#label(input, layout: true, crop: false, **opts) ⇒ Object



69
70
71
72
73
74
75
# File 'lib/anystyle/finder.rb', line 69

def label(input, layout: true, crop: false, **opts)
  dataset = prepare(input, layout: layout, crop: crop, **opts)
  output = model.label(dataset, **opts)
  Wapiti::Dataset.new(dataset.map.with_index { |doc, idx|
    doc.label(output[idx])
  })
end