Module: LearnKit::NaiveBayes::Gaussian::Predict

Included in:
LearnKit::NaiveBayes::Gaussian
Defined in:
lib/learn_kit/naive_bayes/gaussian/predict.rb

Overview

Predict new data

Instance Method Summary collapse

Instance Method Details

#predict(feature_values) ⇒ Object

returns prediction of label for feature input data

f1, f2, f3

output data :label2



11
12
13
14
15
16
 
# File 'lib/learn_kit/naive_bayes/gaussian/predict.rb', line 11

def predict(feature_values)
  res = labels.sort_by do |label_name|
    class_probability(feature_values, label_name)
  end
  res[-1]
end

#predict_proba(feature_values, result = {}) ⇒ Object

returns labels with probabilities input data

f1, f2, f3

output data { label1: 0.01, label2: 0.02 }



23
24
25
26
27
28
 
# File 'lib/learn_kit/naive_bayes/gaussian/predict.rb', line 23

def predict_proba(feature_values, result = {})
  labels.each do |label_name|
    result[label_name] = class_probability(feature_values, label_name)
  end
  result
end