ruby-dnn 0.6.4 → 0.6.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: e7500fad8921cbb1efca407a3dac06aee005257e96403654a0047acd1f05bd91
-  data.tar.gz: 6b2177271b189de689130e75848b92fe747c9b995cc90d16fe2d656102753fb7
+  metadata.gz: e6eff3fa43a3455764d626d75f80d031d747f2806578b14b0082e0d8516534fa
+  data.tar.gz: b32aa37fa92345ffbcc9060eb42e38407ed6d8177594b7d8cebf3993a4a2128a
 SHA512:
-  metadata.gz: dba263057f0ac42d515bfac8ba26c33898329d3ab7e29132f498110eaad22660a2ac8e1b2e9f537ed226c5da07561332ca327123857941598ea7f666fb2d2392
-  data.tar.gz: 3a771401278c68385c9e4a163caee00ffa380ac71ca545ccb500b503c6a92afefb8ca93aa4216dd6783cf10c5187eaa353b916618879e8275b3997a479926117
+  metadata.gz: 070423c94ce2f1a5ee98270517972ca520e9a69073bd8561728e92735fd7da5945f9a8492a43fb51567bba6c00631d5acb9f657696458ca0b21f721db7214225
+  data.tar.gz: 84c18484059187eb9e8da2642a95694048a5413ba444a9eac6685e9b6eb3d585388c857b1efcb78a2c4bb27eb5ee1846b4fb0f9cbd455ab765dd593c72633a5e

data/lib/dnn/core/initializers.rb

@@ -46,16 +46,16 @@ module DNN
     
     class Xavier < Initializer
       def init_param(layer, param_key)
-        prev_dim = layer.prev_layer.shape.reduce(:*)
-        super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(prev_dim))
+        num_prev_nodes = layer.prev_layer.shape.reduce(:*)
+        super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(num_prev_nodes))
       end
     end
     
     
     class He < Initializer
       def init_param(layer, param_key)
-        prev_dim = layer.prev_layer.shape.reduce(:*)
-        super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(prev_dim) * Math.sqrt(2))
+        num_prev_nodes = layer.prev_layer.shape.reduce(:*)
+        super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(num_prev_nodes) * Math.sqrt(2))
       end
     end
 

data/lib/dnn/core/model.rb

@@ -147,7 +147,7 @@ module DNN
     end
   
     def accurate(x, y, batch_size = 1, &batch_proc)
-      batch_size = batch_size >= x.shape[0] ? batch_size : x.shape[0]
+      batch_size = batch_size >= x.shape[0] ? x.shape[0] : batch_size
       correct = 0
       (x.shape[0].to_f / batch_size).ceil.times do |i|
         x_batch = Xumo::SFloat.zeros(batch_size, *x.shape[1..-1])

data/lib/dnn/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "0.6.4"
+  VERSION = "0.6.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 0.6.4
+  version: 0.6.5
 platform: ruby
 authors:
 - unagiootoro
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-08-20 00:00:00.000000000 Z
+date: 2018-08-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray