ruby-dnn 0.6.4 → 0.6.5
Sign up to get free protection for your applications and to get access to all the features.
- 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
|