class Rlid::NaiveBayesGuesser

Public Class Methods

new(default=1) click to toggle source
# 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

Public Instance Methods

guess_language(string) click to toggle source
# 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