Module: LearnKit::NaiveBayes::Gaussian::Fit

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

Overview

Fit test data

Instance Method Summary collapse

Instance Method Details

#fit ⇒ Object

output data { label1: [1, fs_mean1: 2, fs_var1: 3, .., ..], label2: [1, fs_mean1: 2, fs_var1: 3, .., ..] }



10
11
12
13
14
15
16
17
18
19
20
21
22
 
# File 'lib/learn_kit/naive_bayes/gaussian/fit.rb', line 10

def fit
  labels.each do |label_name|
    fit_results[label_name] = []
    (0...feature_size).each do |index|
      fs = feature_set(index, label_name)
      # statistical properties of the feature set
      fs_std = fs.standard_deviation
      fs_mean = fs.mean
      fs_var = fs.variance
      fit_results[label_name] << { fs_std: fs_std, fs_mean: fs_mean, fs_var: fs_var }
    end
  end
end