Class: Bae::NativeClassifier

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

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize ⇒ NativeClassifier

Returns a new instance of NativeClassifier.



9
10
11
 
# File 'lib/bae/native_classifier.rb', line 9

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

Instance Attribute Details

#internal_classifier ⇒ Object (readonly)

Returns the value of attribute internal_classifier.



7
8
9
 
# File 'lib/bae/native_classifier.rb', line 7

def internal_classifier
  @internal_classifier
end

Instance Method Details

#classify(feature) ⇒ Object 



17
18
19
 
# File 'lib/bae/native_classifier.rb', line 17

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

#finish_training! ⇒ Object 



21
22
23
 
# File 'lib/bae/native_classifier.rb', line 21

def finish_training!
  internal_classifier.calculateInitialLikelihoods()
end

#train(label, feature) ⇒ Object 



13
14
15
 
# File 'lib/bae/native_classifier.rb', line 13

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