Class: Ocarina::Network

Inherits:
Object
  • Object
show all
Includes:
Util
Defined in:
lib/ocarina/network.rb

Overview

a network of neurons

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods included from Util

#char_to_binary_string, #filename_for_noise_image, #filename_for_quantized_image, #filename_for_training_image, #int_to_binary_string, #is_lower?, #is_upper?, #noise_image_for_char, #pixel_number_to_col, #pixel_number_to_row, #pixel_to_bit, #quantize_image, #reference_image_for_char, #sigma

Constructor Details

#initialize(config) ⇒ Network

Returns a new instance of Network.



13
14
15
16
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/ocarina/network.rb', line 13

def initialize(config)

  @config         = config
  @num_inputs     = config.num_inputs  # total of bits in the image

  #@hidden_count   = (1.5 * num_inputs).to_i  # somewhat arbitrary
  @hidden_count   = 25

  @input_values   = []   # image bits
  @input_weights  = []   # weights from inputs -> hidden nodes

  @hidden_outputs = []   # after feed-forward, what the hidden nodes output

  @output_weights = []   # weights from hidden nodes -> output nodes
  @output_values  = []   # after feed-forward, what the output nodes output

  @output_errors  = []
  @hidden_errors  = []

  assign_random_weights

  #puts "@input_weights: #{@input_weights}"

  total_input_weights = @input_weights.map { |array| array.reduce(:+) }.reduce(:+)
  puts "total_input_weights: #{total_input_weights}"

  total_output_weights = @output_weights.map { |array| array.reduce(:+) }.reduce(:+)
  puts "total_output_weights: #{total_output_weights}"
end

Instance Attribute Details

#config ⇒ Object

Returns the value of attribute config.



11
12
13
 
# File 'lib/ocarina/network.rb', line 11

def config
  @config
end

#current_error ⇒ Object

Returns the value of attribute current_error.



11
12
13
 
# File 'lib/ocarina/network.rb', line 11

def current_error
  @current_error
end

Class Method Details

.load_network_from_file(filepath) ⇒ Object

load a previously-trained network



101
102
103
104
105
 
# File 'lib/ocarina/network.rb', line 101

def self.load_network_from_file(filepath)
  File.open(filepath) do |file|
    Marshal.load(file)
  end
end

Instance Method Details

#recognize(image) ⇒ Object

Attempt to recognize the character displayed on the given image. image should be an instance of Magick::Image.

Returns the integer ASCII code for the recognized character.



48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
 
# File 'lib/ocarina/network.rb', line 48

def recognize(image)
  # quantize to two-color
  image = quantize_image(image)

  # the binary string we expect to see from the output nodes
  assign_inputs image

  calculate_hidden_outputs
  calculate_final_outputs

  #@output_values.each.with_index { |v, i| puts "index: #{i} => #{v}" }

  # process results
  #
  binary_string = quantized_result.inject("") { |accum, bit| "#{accum}#{bit.to_s}" }
  binary_string.to_i(2)
end

#save_network_to_file(filepath) ⇒ Object

persist the network



93
94
95
96
97
 
# File 'lib/ocarina/network.rb', line 93

def save_network_to_file(filepath)
  File.open(filepath,'w') do|file|
    Marshal.dump(self, file)
  end
end

#train(image, target_char) ⇒ Object

Train the network on the image, using target_char as the expected result.

image should be an instance of Magick::Image.



70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
 
# File 'lib/ocarina/network.rb', line 70

def train(image, target_char)
  # quantize to two-color
  image = quantize_image(image)
  #image.write(filename_for_quantized_image(target_char, 'gif'))

  # the binary string we expect to see from the output nodes
  @target_binary_string = char_to_binary_string(target_char)

  assign_inputs image

  calculate_hidden_outputs
  calculate_final_outputs
  calculate_output_errors
  calculate_hidden_errors

  # process results
  #
  adjust_output_weights
  adjust_input_weights
end