Class: Trainer

Inherits:
TinyClassifierBase show all
Defined in:
lib/tiny-classifier/trainer.rb

Constant Summary

Constants inherited from TinyClassifierBase

TinyClassifierBase::TOKENIZERS

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from TinyClassifierBase

#parse_command_line_options

Constructor Details

#initialize ⇒ Trainer

Returns a new instance of Trainer.



28
29
30
31
 
# File 'lib/tiny-classifier/trainer.rb', line 28

def initialize
  super
  option_parser.banner += " LABEL"
end

Class Method Details

.run(argv = nil) ⇒ Object 



20
21
22
23
24
25
 
# File 'lib/tiny-classifier/trainer.rb', line 20

def run(argv=nil)
  argv ||= ARGV.dup
  trainer = new
  *labels = trainer.parse_command_line_options(argv)
  trainer.run(label: labels.first)
end

Instance Method Details

#run(params) ⇒ Object 



33
34
35
36
37
38
39
40
41
42
43
44
 
# File 'lib/tiny-classifier/trainer.rb', line 33

def run(params)
  @label = params[:label]
  prepare_label
  if input.empty?
    STDERR.puts("Error: No effective input.")
    false
  else
    classifier.send("train_#{@label}", input)
    save
    true
  end
end