Class: Company::Mapping::CompanyCorpus

Inherits:
Corpus
  • Object
show all
Defined in:
lib/company/mapping/document_utils/company_corpus.rb

Instance Method Summary collapse

Constructor Details

#initialize(path = nil) ⇒ CompanyCorpus 



4
5
6
7
 
# File 'lib/company/mapping/document_utils/company_corpus.rb', line 4

def initialize(path=nil)
  super()
  import_csv path if path
end

Instance Method Details

#add(id, company_name, aliases = []) ⇒ Object 



9
10
11
12
13
14
 
# 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) ⇒ Object

build a corpus from a csv file



17
18
19
20
21
22
23
 
# 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