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

  end
end

Version data entries

6 entries across 6 versions & 1 rubygems

Version	Path
bae-0.0.6-java	lib/bae/classifier.rb
bae-0.0.5-java	lib/bae/classifier.rb
bae-0.0.4-java	lib/bae/classifier.rb
bae-0.0.3-java	lib/bae/classifier.rb
bae-0.0.2-java	lib/bae/classifier.rb
bae-0.0.1	lib/bae/classifier.rb