class LearnKit::NaiveBayes::Gaussian

Gaussian NB

Attributes

data_set[R]
feature_size[R]
fit_results[R]
labels[R]

Public Class Methods

new(args = {}) click to toggle source

input data { label1: [[f1, f2, f3], [f1, f2, f3]], label2: [[f4, f5, f6], [f4, f5, f6]] }

# File lib/learn_kit/naive_bayes/gaussian.rb, line 17
def initialize(args = {})
  @data_set = args[:data_set]
  @labels = data_set.keys
  @feature_size = data_set.values.first[0].size
  @fit_results = {}
end