cerebrum 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 8ad9f879fedc9edbdb4898c23ac34c774ce96d93
-  data.tar.gz: e69a64b6c1cb78e474f7223cebc3879588511e37
+  metadata.gz: 822cf99b99f8c49d058c2d1ff99fec93cc910228
+  data.tar.gz: 0ace6aab3806f1344a73af0300c25992e8753d96
 SHA512:
-  metadata.gz: 5f0fafaae68fd966304f3ef83b8a8b4b124931132693ffe2a53607f37c3f1b79966d9f32172b093200467c9801cff505c14bac8af3a668f1d88b6c71e432c4ab
-  data.tar.gz: d0fc8ce6021dc2262cb6e4bb12f34aceb1fae9408aeb4b0e3918ecf12f2f557217d1bf07082fa5fae8153108ca8fc17562df474fa30e55c4152ba1ac4b3a2092
+  metadata.gz: ceca1caa39e8bc1866de84ef9047c431badc913cbaa8c178d78426095e4455d21727d03e03a28b2a5dcdc1b82c33865e9f7a26f147e6ce175785455a0d731e8a
+  data.tar.gz: 3e1e98735a3731120e029bf13aeaeb78469c89c7c39d941805bf9ccc93de61272a013e5f61aaff160c1691979fb75f63059b8f20dadf9d9befef9c5710095f1a

data/README.md

@@ -62,7 +62,7 @@ network.train([
 ]);
 
 result = network.run({ r: 1, g: 0.4, b: 0 })
-# => { :black=>0.011967728530458011, :white=>0.9871010273923573 }
+# => { black: 0.011967728530458011, white: 0.9871010273923573 }
 ```
 
 #### Cerebrum Options
@@ -115,6 +115,22 @@ The output of `Cerebrum#train` is a hash of information about how the training w
 network.train(data, options)
 # => { error: 0.005324233132423, iterations: 9001 }
 ```
+
+#### JSON
+
+Serialize or load in the state of a trained network with JSON:
+
+``` ruby
+saved_state = network.save_state
+nn = Cerebrum.new
+nn.load_state(saved_state)
+
+nn.run({ r: 1, g: 0.4, b: 0 })
+# => { black: 0.011967728530458011, white: 0.9871010273923573 }
+```
+
+Additionally the new network can be separately trained whilst retaining all
+previous training from the other network.
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run

data/lib/cerebrum/cerebrum.rb

@@ -1,5 +1,6 @@
 require_relative "data_scrubber"
 require_relative "cerebrum_helper"
+require 'json'
 
 class Cerebrum
   include CerebrumHelper
@@ -25,8 +26,8 @@ class Cerebrum
   end
 
   def train(training_set, options = Hash.new)
-    @input_lookup_table ||= get_input_lookup_table(training_set)
-    @output_lookup_table ||= get_output_lookup_table(training_set)
+    @input_lookup_table   ||= get_input_lookup_table(training_set)
+    @output_lookup_table  ||= get_output_lookup_table(training_set)
     training_set = scrub_dataset(training_set)
 
     iterations        = options[:iterations] || 20000
@@ -62,6 +63,44 @@ class Cerebrum
     @output_lookup_table ? to_features_given_vector(output, @output_lookup_table) : output
   end
 
+  def save_state
+    {
+      biases: @biases,
+      binary_thresh: @binary_thresh,
+      changes: @changes,
+      deltas: @deltas,
+      errors: @errors,
+      hidden_layers: @hidden_layers,
+      input_lookup_table: @input_lookup_table,
+      layer_sizes: @layer_sizes,
+      layers: @layers,
+      learning_rate: @learning_rate,
+      momentum: @momentum,
+      output_lookup_table: @output_lookup_table,
+      outputs: @outputs,
+      weights: @weights
+    }.to_json
+  end
+
+  def load_state(saved_state)
+    state = JSON.parse(saved_state, symbolize_names: true)
+
+    @biases               = state[:biases]
+    @binary_thresh        = state[:binary_thresh]
+    @changes              = state[:changes]
+    @deltas               = state[:deltas]
+    @errors               = state[:errors]
+    @hidden_layers        = state[:hidden_layers]
+    @input_lookup_table   = state[:input_lookup_table]
+    @layer_sizes          = state[:layer_sizes]
+    @layers               = state[:layers]
+    @learning_rate        = state[:learning_rate]
+    @momentum             = state[:momentum]
+    @output_lookup_table  = state[:output_lookup_table]
+    @outputs              = state[:outputs]
+    @weights              = state[:weights]
+  end
+
   private
 
   def construct_network(layer_sizes)

data/lib/cerebrum/version.rb

@@ -1,3 +1,3 @@
 class Cerebrum
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: cerebrum
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Irfan Sharif
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2016-05-22 00:00:00.000000000 Z
+date: 2016-07-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler