1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
# Author:: David Fayram (mailto:dfayram@lensmen.net)
# Copyright:: Copyright (c) 2005 David Fayram II
# License:: LGPL
module ClassifierReborn
# This class keeps a word => index mapping. It is used to map stemmed words
# to dimensions of a vector.
class WordList
def initialize
@location_table = {}
end
# Adds a word (if it is new) and assigns it a unique dimension.
def add_word(word)
@location_table[word] = @location_table.size unless @location_table[word]
end
# Returns the dimension of the word or nil if the word is not in the space.
def [](lookup)
@location_table[lookup]
end
def word_for_index(ind)
@location_table.invert[ind]
end
# Returns the number of words mapped.
def size
@location_table.size
end
end
end
|
