Class: LearnKit::NaiveBayes::Gaussian

Inherits:
Object
  • Object
show all
Includes:
Fit, Predict, Score
Defined in:
lib/learn_kit/naive_bayes/gaussian.rb,
lib/learn_kit/naive_bayes/gaussian/fit.rb,
lib/learn_kit/naive_bayes/gaussian/score.rb,
lib/learn_kit/naive_bayes/gaussian/predict.rb

Overview

Gaussian NB

Defined Under Namespace

Modules: Fit, Predict, Score

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods included from Score

#score

Methods included from Predict

#predict, #predict_proba

Methods included from Fit

#fit

Constructor Details

#initialize(args = {}) ⇒ Gaussian

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



17
18
19
20
21
22
 
# 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

Instance Attribute Details

#data_set ⇒ Object (readonly)

Returns the value of attribute data_set.



13
14
15
 
# File 'lib/learn_kit/naive_bayes/gaussian.rb', line 13

def data_set
  @data_set
end

#feature_size ⇒ Object (readonly)

Returns the value of attribute feature_size.



13
14
15
 
# File 'lib/learn_kit/naive_bayes/gaussian.rb', line 13

def feature_size
  @feature_size
end

#fit_results ⇒ Object (readonly)

Returns the value of attribute fit_results.



13
14
15
 
# File 'lib/learn_kit/naive_bayes/gaussian.rb', line 13

def fit_results
  @fit_results
end

#labels ⇒ Object (readonly)

Returns the value of attribute labels.



13
14
15
 
# File 'lib/learn_kit/naive_bayes/gaussian.rb', line 13

def labels
  @labels
end