Class: Bayes::PlainBayes

Inherits:
FilterBase show all
Defined in:
lib/bayes.rb

Defined Under Namespace

Classes: Corpus

Instance Attribute Summary

Attributes inherited from FilterBase

#charset, #db_name, #ham, #spam

Instance Method Summary collapse

Methods inherited from FilterBase

#[], #convert, #initialize, #save

Constructor Details

This class inherits a constructor from Bayes::FilterBase

Instance Method Details

#estimate(tokens, take = 15) ⇒ Object 



191
192
193
194
195
196
197
 
# File 'lib/bayes.rb', line 191

def estimate(tokens, take=15)
	s = tokens.uniq.map{|i| score(i)}.compact.sort{|a, b| (0.5-a).abs <=> (0.5-b)}.reverse[0...take]
	return nil if s.empty? || s.include?(1.0) && s.include?(0.0)

	prod = s.inject(1.0){|r, i| r*i}
	return prod/(prod+s.inject(1.0){|r, i| r*(1-i)})
end

#score(token) ⇒ Object 



184
185
186
187
188
189
 
# File 'lib/bayes.rb', line 184

def score(token)
	return nil unless @spam.include?(token) || @ham.include?(token)
	s = @spam[token]
	h = @ham[token]
	s/(s+h)
end