Merge pull request #14 from modern-fortran/12-loading-network-problems

add example that shows saving a trained network to a file, then loadi…

Author: milancurcic

CMakeLists.txt

@@ -78,7 +78,7 @@ foreach(execid mnist network_save network_sync set_activation_function)
   add_test(test_${execid} bin/test_${execid})
 endforeach()
 
-foreach(execid mnist simple sine)
+foreach(execid mnist save_and_load simple sine)
   add_executable(example_${execid} src/tests/example_${execid}.f90)
   target_link_libraries(example_${execid} neural ${LIBS})
   add_test(example_${execid} bin/example_${execid})

src/tests/example_save_and_load.f90

@@ -0,0 +1,32 @@
+program example_save_and_load
+
+  use mod_network, only: network_type
+  implicit none
+
+  type(network_type) :: net1, net2
+  real, allocatable :: input(:), output(:)
+  integer :: i
+
+  net1 = network_type([3, 5, 2])
+
+  input = [0.2, 0.4, 0.6]
+  output = [0.123456, 0.246802]
+
+  ! train network 1
+  do i = 1, 500
+    call net1 % train(input, output, eta=1.0)
+  end do
+
+  ! save network 1 to file
+  call net1 % save('my_simple_net.txt')
+
+  ! load network 2 from file 
+  !net2 = network_type([3, 5, 2])
+  call net2 % load('my_simple_net.txt')
+  call net2 % set_activation('sigmoid')
+
+  print *, 'Network 1 output: ', net1 % output(input) 
+  print *, 'Network 2 output: ', net2 % output(input)
+  print *, 'Outputs match: ', all(net1 % output(input) == net2 % output(input))
+
+end program example_save_and_load