module LearnKit::NaiveBayes::Gaussian::Score

Calc accuracy for test data

Public Instance Methods

score(total = 0, correct = 0) click to toggle source

calculate accuracy for test data

# File lib/learn_kit/naive_bayes/gaussian/score.rb, line 7
def score(total = 0, correct = 0)
  labels.each do |label_name|
    data_set[label_name].each do |feature|
      predicted_label = predict(feature)
      correct += 1 if predicted_label == label_name
      total += 1
    end
  end
  total.zero? ? 0 : (correct * 100.0 / total).round(4)
end