class Company::Mapping::CompanyCorpus
Public Class Methods
new(path=nil)
click to toggle source
Calls superclass method
# File lib/company/mapping/document_utils/company_corpus.rb, line 4 def initialize(path=nil) super() import_csv path if path end
Public Instance Methods
add(id, company_name, aliases=[])
click to toggle source
# File lib/company/mapping/document_utils/company_corpus.rb, line 9 def add id, company_name, aliases=[] push doc(company_name, id) aliases.each_with_index do |company_alias, i| push doc(company_alias, "#{id}_#{i}") end end
import_csv(path)
click to toggle source
build a corpus from a csv file
# File lib/company/mapping/document_utils/company_corpus.rb, line 17 def import_csv path CSV.foreach(path) do |row| array = row.first.split(";") add array.first, array[1], array[2..-1] end @corpus end
Private Instance Methods
doc(content, id)
click to toggle source
# File lib/company/mapping/document_utils/company_corpus.rb, line 27 def doc content, id alias_doc = TextDocument.new alias_doc.contents = content.gsub(",", "").gsub(".", "") alias_doc.id = id alias_doc end