Contents

module Bae
  class Classifier

    attr_reader :internal_classifier

    def initialize
      @internal_classifier = ::Java::Bae::NaiveBayesClassifier.new
    end

    def train(label, feature)
      internal_classifier.train(label, ::Java::Bae::Document.new(feature))
    end

    def classify(feature)
      internal_classifier.classify(::Java::Bae::Document.new(feature))
    end

    def finish_training!
      internal_classifier.calculateInitialLikelihoods()
    end

  end
end

Version data entries

1 entries across 1 versions & 1 rubygems

Version	Path
bae-0.0.7-java	lib/bae/classifier.rb