Module: Brainz::Algorithms::Backpropagation

Defined in:
lib/brainz/backpropagation.rb

Instance Method Summary collapse

Instance Method Details

#fix_weights(targets) ⇒ Object 



12
13
14
 
# File 'lib/brainz/backpropagation.rb', line 12

def fix_weights(targets)
  @network.fix_weights(targets)
end

#update(input) ⇒ Object 



4
5
6
7
8
9
10
 
# File 'lib/brainz/backpropagation.rb', line 4

def update(input)
  @network ||= ::Brainz::Network.new(
      @num_input, [num_hidden, @num_output * 2], @num_output, momentum: momentum, learning_rate: learning_rate
  )
  @network.update(input)
  self
end