Contents

require_relative 'gaussian/fit'
require_relative 'gaussian/predict'
require_relative 'gaussian/score'

module LearnKit
  module NaiveBayes
    # Gaussian NB
    class Gaussian
      include LearnKit::NaiveBayes::Gaussian::Fit
      include LearnKit::NaiveBayes::Gaussian::Predict
      include LearnKit::NaiveBayes::Gaussian::Score

      attr_reader :data_set, :labels, :feature_size, :fit_results

      # input data
      # { label1: [[f1, f2, f3], [f1, f2, f3]], label2: [[f4, f5, f6], [f4, f5, f6]] }
      def initialize(args = {})
        @data_set = args[:data_set]
        @labels = data_set.keys
        @feature_size = data_set.values.first[0].size
        @fit_results = {}
      end
    end
  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
learn_kit-0.0.1	lib/learn_kit/naive_bayes/gaussian.rb