Class: Treat::Workers::Learners::Classifiers::MLP

Inherits:
Object
  • Object
show all
Defined in:
lib/treat/workers/learners/classifiers/mlp.rb

Overview

Classification based on a multilayer perceptron.

Constant Summary collapse

DefaultOptions = 
{
  num_inputs: 3,
  hidden_neurons: [2, 8, 4, 3, 4],
  num_outputs: 1,
  max_neurons: 1000,
  neurons_between_reports: 1,
  desired_error: 0.1
}
@@classifiers = 
{}

Class Method Summary collapse

Class Method Details

.classify(entity, options = {}) ⇒ Object 



17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
 
# File 'lib/treat/workers/learners/classifiers/mlp.rb', line 17

def self.classify(entity, options = {})
  options = DefaultOptions.merge(options)
  dset = options[:training]
  prob, items = dset.problem, dset.items
  if !@@classifiers[prob]
    fann = RubyFann::Standard.new(options)
    inputs = items.map { |it| it[:features][0..-2] }
    outputs = items.map { |it| [it[:features][-1]] }
    training = silence_stdout do 
      RubyFann::TrainData.new(inputs: 
      inputs, desired_outputs: outputs)
    end
    params = [options[:max_neurons],
    options[:neurons_between_reports],
    options[:desired_error]]
    fann.train_on_data(training, *params)
    @@classifiers[prob] = fann
  else
    fann = @@classifiers[prob]
  end
  vect = prob.export_features(entity, false)
  Treat.core.verbosity.silence ? 
  silence_stdout { fann.run(vect)[0] } :
  fann.run(vect)[0]
end