Sha256: 539e184e01ee709f5f5ee64cd1d28c9a0c51cb0135f6f39f2473475316086ddb
Contents?: true
Size: 1.14 KB
Versions: 13
Compression:
Stored size: 1.14 KB
Contents
module Ankusa
class MemoryStorage
def initialize
init_tables
end
def classnames
@total_doc_counts.keys
end
def reset
init_tables
end
def drop_tables
end
def init_tables
@freqs = {}
@total_word_counts = Hash.new(0)
@total_doc_counts = Hash.new(0)
@klass_word_counts = {}
@klass_doc_counts = {}
end
def get_vocabulary_sizes
count = Hash.new 0
@freqs.each { |w, ks|
ks.keys.each { |k| count[k] += 1 }
}
count
end
def get_word_counts(word)
@freqs.fetch word, Hash.new(0)
end
def get_total_word_count(klass)
@total_word_counts[klass]
end
def get_doc_count(klass)
@total_doc_counts[klass]
end
def incr_word_count(klass, word, count)
@freqs[word] ||= Hash.new(0)
@freqs[word][klass] += count
end
def incr_total_word_count(klass, count)
@total_word_counts[klass] += count
end
def incr_doc_count(klass, count)
@total_doc_counts[klass] += count
end
def doc_count_totals
@total_doc_counts
end
def close
end
end
end
Version data entries
13 entries across 13 versions & 1 rubygems