Class: Rlid::NaiveBayesGuesser

Inherits:
LanguageGuesser show all
Defined in:
lib/rlid/language_guesser/naive_bayes_guesser.rb

Direct Known Subclasses

NaiveBayesProbabilityGuesser

Instance Attribute Summary

Attributes inherited from LanguageGuesser

#name

Instance Method Summary collapse

Constructor Details

#initialize(default = 1) ⇒ NaiveBayesGuesser

Returns a new instance of NaiveBayesGuesser.



8
9
10
11
12
13
14
 
# File 'lib/rlid/language_guesser/naive_bayes_guesser.rb', line 8

def initialize(default=1)
  #print "Naive Bayes: loading models.."
  @models = NaiveBayesModels.load
  @models.default_count = default
  @name = "Naive Bayes"
  #puts " Done!"
end

Instance Method Details

#guess_language(string) ⇒ Object 



16
17
18
19
20
21
22
23
24
25
26
 
# File 'lib/rlid/language_guesser/naive_bayes_guesser.rb', line 16

def guess_language(string)
  max_prob = 0.0
  best_language = nil
  @models.probabilities(string) do |lang, prob|
    if prob > max_prob
      max_prob = prob
      best_language = lang
    end
  end
  best_language
end