ruby-dnn 0.8.2 → 0.8.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ee33bcc8216f37bee65157d0d7330664e46b23da56100b0cb823254690cf4716
-  data.tar.gz: e6448eed2fd19dbdabe9ebf03fe2c4faf04af7c97e0e4acbba4051970c5a9a15
+  metadata.gz: 9c34f355a4fdde6dce0cdc8c84f0eb3c670d61aa939b6e33b3c1b21b96ff84e2
+  data.tar.gz: 27362633d8824d7b5a1e765a50c5f248c3bc680ee9261a0cb2e103fde9eca42c
 SHA512:
-  metadata.gz: b8d1e6e7d895b993157e5fee5e02650cb201801f7681b8a4b05267fc346c2bf3088eaadb8a0dcea97e7096cd8a3c942855a704b775e2748f8c4e20747efda1d0
-  data.tar.gz: ae8c5a109c148eec1f20c723c49b49eb223023f7e25bcf3b852cb208c210e6a9262a838824eb58d175259db99676aaa4b0d1fd7ee4316ee4d044f3e193236702
+  metadata.gz: 3fac1e61fffaca5126c4caa488bdadc02d613bd88a368316ce7924b162394d9cc1d9e0b4f8ec6b56078cea5cb23d02e89bed95ac0b0bba4c374290034a3550c2
+  data.tar.gz: 4df5ad737f827fc5ed063be6cc8d22bee34ef95e35c9c5522dc69b7bd453658b0af04b0f6adb1bb7c94d4f52f16d44040ee1097f83e533fa01717f1736a1a8f6

data/lib/dnn/core/layers.rb

@@ -20,13 +20,13 @@ module DNN
 
       # Forward propagation.
       # Classes that inherit from this class must implement this method.
-      def forward
+      def forward(x)
         raise NotImplementedError.new("Class '#{self.class.name}' has implement method 'forward'")
       end
 
       # Backward propagation.
       # Classes that inherit from this class must implement this method.
-      def backward
+      def backward(dout)
         raise NotImplementedError.new("Class '#{self.class.name}' has implement method 'update'")
       end
     
@@ -181,8 +181,8 @@ module DNN
       end
     
       def initialize(num_nodes,
-                     weight_initializer: nil,
-                     bias_initializer: nil,
+                     weight_initializer: Initializers::RandomNormal.new,
+                     bias_initializer: Initializers::Zeros.new,
                      l1_lambda: 0,
                      l2_lambda: 0)
         super(weight_initializer: weight_initializer, bias_initializer: bias_initializer,
@@ -328,16 +328,6 @@ module DNN
       def initialize(momentum: 0.9)
         super()
         @momentum = momentum
-        @params[:gamma] = @gamma = LearningParam.new
-        @params[:beta] = @beta = LearningParam.new
-        @params[:running_mean] = nil
-        @params[:running_var] = nil
-      end
-
-      def build(model)
-        super
-        @params[:running_mean] ||= Xumo::SFloat.zeros(*shape)
-        @params[:running_var] ||= Xumo::SFloat.zeros(*shape)
       end
 
       def forward(x)
@@ -377,8 +367,12 @@ module DNN
       private
     
       def init_params
+        @params[:gamma] = @gamma = LearningParam.new
+        @params[:beta] = @beta = LearningParam.new
         @gamma.data = Xumo::SFloat.ones(*shape)
         @beta.data = Xumo::SFloat.zeros(*shape)
+        @params[:running_mean] = Xumo::SFloat.zeros(*shape)
+        @params[:running_var] = Xumo::SFloat.zeros(*shape)
       end
     end
   end

data/lib/dnn/core/rnn_layers.rb

@@ -14,8 +14,8 @@ module DNN
       def initialize(num_nodes,
                      stateful: false,
                      return_sequences: true,
-                     weight_initializer: nil,
-                     bias_initializer: nil,
+                     weight_initializer: Initializers::RandomNormal.new,
+                     bias_initializer: Initializers::Zeros.new,
                      l1_lambda: 0,
                      l2_lambda: 0)
         super(weight_initializer: weight_initializer, bias_initializer: bias_initializer,
@@ -430,7 +430,7 @@ module DNN
                      stateful: false,
                      return_sequences: true,
                      weight_initializer: Initializers::RandomNormal.new,
-                     bias_initializer: Initializers::Zeros.new
+                     bias_initializer: Initializers::Zeros.new,
                      l1_lambda: 0,
                      l2_lambda: 0)
         super

data/lib/dnn/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "0.8.2"
+  VERSION = "0.8.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 0.8.2
+  version: 0.8.3
 platform: ruby
 authors:
 - unagiootoro
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-02-06 00:00:00.000000000 Z
+date: 2019-02-18 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray