Class: Bae::Classifier

Inherits:
Object
  • Object
show all
Defined in:
lib/bae/classifier.rb

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ Classifier

Returns a new instance of Classifier.



6
7
8
 
# File 'lib/bae/classifier.rb', line 6

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

Instance Attribute Details

#internal_classifier ⇒ Object (readonly)

Returns the value of attribute internal_classifier.



4
5
6
 
# File 'lib/bae/classifier.rb', line 4

def internal_classifier
  @internal_classifier
end

Instance Method Details

#classify(feature) ⇒ Object 



14
15
16
 
# File 'lib/bae/classifier.rb', line 14

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

#finish_training! ⇒ Object 



18
19
20
 
# File 'lib/bae/classifier.rb', line 18

def finish_training!
  internal_classifier.calculateInitialLikelihoods()
end

#train(label, feature) ⇒ Object 



10
11
12
 
# File 'lib/bae/classifier.rb', line 10

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