ruby-dnn 0.6.4 → 0.6.5
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/lib/dnn/core/initializers.rb +4 -4
- data/lib/dnn/core/model.rb +1 -1
- data/lib/dnn/version.rb +1 -1
- metadata +2 -2
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: e6eff3fa43a3455764d626d75f80d031d747f2806578b14b0082e0d8516534fa
|
|
4
|
+
data.tar.gz: b32aa37fa92345ffbcc9060eb42e38407ed6d8177594b7d8cebf3993a4a2128a
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: 070423c94ce2f1a5ee98270517972ca520e9a69073bd8561728e92735fd7da5945f9a8492a43fb51567bba6c00631d5acb9f657696458ca0b21f721db7214225
|
|
7
|
+
data.tar.gz: 84c18484059187eb9e8da2642a95694048a5413ba444a9eac6685e9b6eb3d585388c857b1efcb78a2c4bb27eb5ee1846b4fb0f9cbd455ab765dd593c72633a5e
|
|
@@ -46,16 +46,16 @@ module DNN
|
|
|
46
46
|
|
|
47
47
|
class Xavier < Initializer
|
|
48
48
|
def init_param(layer, param_key)
|
|
49
|
-
|
|
50
|
-
super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(
|
|
49
|
+
num_prev_nodes = layer.prev_layer.shape.reduce(:*)
|
|
50
|
+
super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(num_prev_nodes))
|
|
51
51
|
end
|
|
52
52
|
end
|
|
53
53
|
|
|
54
54
|
|
|
55
55
|
class He < Initializer
|
|
56
56
|
def init_param(layer, param_key)
|
|
57
|
-
|
|
58
|
-
super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(
|
|
57
|
+
num_prev_nodes = layer.prev_layer.shape.reduce(:*)
|
|
58
|
+
super(layer, param_key, layer.params[param_key].rand_norm / Math.sqrt(num_prev_nodes) * Math.sqrt(2))
|
|
59
59
|
end
|
|
60
60
|
end
|
|
61
61
|
|
data/lib/dnn/core/model.rb
CHANGED
|
@@ -147,7 +147,7 @@ module DNN
|
|
|
147
147
|
end
|
|
148
148
|
|
|
149
149
|
def accurate(x, y, batch_size = 1, &batch_proc)
|
|
150
|
-
batch_size = batch_size >= x.shape[0] ?
|
|
150
|
+
batch_size = batch_size >= x.shape[0] ? x.shape[0] : batch_size
|
|
151
151
|
correct = 0
|
|
152
152
|
(x.shape[0].to_f / batch_size).ceil.times do |i|
|
|
153
153
|
x_batch = Xumo::SFloat.zeros(batch_size, *x.shape[1..-1])
|
data/lib/dnn/version.rb
CHANGED
metadata
CHANGED
|
@@ -1,14 +1,14 @@
|
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
|
2
2
|
name: ruby-dnn
|
|
3
3
|
version: !ruby/object:Gem::Version
|
|
4
|
-
version: 0.6.
|
|
4
|
+
version: 0.6.5
|
|
5
5
|
platform: ruby
|
|
6
6
|
authors:
|
|
7
7
|
- unagiootoro
|
|
8
8
|
autorequire:
|
|
9
9
|
bindir: exe
|
|
10
10
|
cert_chain: []
|
|
11
|
-
date: 2018-08-
|
|
11
|
+
date: 2018-08-22 00:00:00.000000000 Z
|
|
12
12
|
dependencies:
|
|
13
13
|
- !ruby/object:Gem::Dependency
|
|
14
14
|
name: numo-narray
|