Merge pull request #8 from hugoabonizio/feat/add-save-and-load

Add save and load methods

Author: bararchy

spec/network_spec.cr

@@ -76,4 +76,46 @@ describe Fann::Network do
     ann.close
     (result < [0.1]).should be_true
   end
+
+  context "saving current network" do
+    it "saves standard networks" do
+      tempfile = Tempfile.new("foo")
+      File.size(tempfile.path).should eq 0
+      ann = Fann::Network::Standard.new(2, [2], 1)
+      ann.save(tempfile.path)
+      (File.size(tempfile.path) > 0).should be_true
+    end
+
+    it "saves cascade networks" do
+      tempfile = Tempfile.new("bar")
+      File.size(tempfile.path).should eq 0
+      ann = Fann::Network::Cascade.new(2, 1)
+      ann.save(tempfile.path)
+      (File.size(tempfile.path) > 0).should be_true
+    end
+  end
+
+  context "loading a configuration file" do
+    it "loads standard networks" do
+      input = 2
+      output = 1
+      tempfile = Tempfile.new("standard")
+      original = Fann::Network::Standard.new(input, [2], output)
+      original.save(tempfile.path)
+      loaded = Fann::Network::Standard.new(tempfile.path)
+      loaded.input_size.should eq input
+      loaded.output_size.should eq output
+    end
+
+    it "loads cascade networks" do
+      input = 2
+      output = 1
+      tempfile = Tempfile.new("cascade")
+      original = Fann::Network::Cascade.new(input, output)
+      original.save(tempfile.path)
+      loaded = Fann::Network::Cascade.new(tempfile.path)
+      loaded.input_size.should eq input
+      loaded.output_size.should eq output
+    end
+  end
 end

spec/spec_helper.cr

@@ -1,2 +1,3 @@
 require "spec"
+require "tempfile"
 require "../src/crystal-fann"

src/crystal-fann/cascade_network.cr

@@ -2,13 +2,21 @@ module Fann
   module Network
     class Cascade
       property :nn
+      getter :input_size
+      getter :output_size
 
       def initialize(input : Int32, output : Int32)
         @output_size = output
         @input_size = input
         @nn = LibFANN.create_shortcut(2, input, output)
       end
 
+      def initialize(path : String)
+        @nn = LibFANN.create_from_file(path)
+        @input_size = LibFANN.get_num_input(@nn)
+        @output_size = LibFANN.get_num_output(@nn)
+      end
+
       def mse
         LibFANN.get_mse(@nn)
       end
@@ -43,6 +51,10 @@ module Fann
         result = LibFANN.run(@nn, input.to_unsafe)
         Slice.new(result, @output_size).to_a
       end
+
+      def save(path : String) : Int32
+        LibFANN.save(@nn, path)
+      end
     end
   end
 end

src/crystal-fann/standard_network.cr

@@ -2,6 +2,8 @@ module Fann
   module Network
     class Standard
       property :nn
+      getter :input_size
+      getter :output_size
 
       def initialize(input : Int32, hidden : Array(Int32), output : Int32)
         @logger = Logger.new(STDOUT)
@@ -17,6 +19,13 @@ module Fann
         @nn = LibFANN.create_standard_array(layers.size, layers.to_unsafe)
       end
 
+      def initialize(path : String)
+        @logger = Logger.new(STDOUT)
+        @nn = LibFANN.create_from_file(path)
+        @input_size = LibFANN.get_num_input(@nn)
+        @output_size = LibFANN.get_num_output(@nn)
+      end
+
       def mse
         LibFANN.get_mse(@nn)
       end
@@ -79,6 +88,10 @@ module Fann
         result = LibFANN.run(@nn, input.to_unsafe)
         Slice.new(result, @output_size).to_a
       end
+
+      def save(path : String) : Int32
+        LibFANN.save(@nn, path)
+      end
     end
   end
 end