ruby-dnn 0.8.2 → 0.8.3
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/layers.rb +8 -14
- data/lib/dnn/core/rnn_layers.rb +3 -3
- 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: 9c34f355a4fdde6dce0cdc8c84f0eb3c670d61aa939b6e33b3c1b21b96ff84e2
|
4
|
+
data.tar.gz: 27362633d8824d7b5a1e765a50c5f248c3bc680ee9261a0cb2e103fde9eca42c
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 3fac1e61fffaca5126c4caa488bdadc02d613bd88a368316ce7924b162394d9cc1d9e0b4f8ec6b56078cea5cb23d02e89bed95ac0b0bba4c374290034a3550c2
|
7
|
+
data.tar.gz: 4df5ad737f827fc5ed063be6cc8d22bee34ef95e35c9c5522dc69b7bd453658b0af04b0f6adb1bb7c94d4f52f16d44040ee1097f83e533fa01717f1736a1a8f6
|
data/lib/dnn/core/layers.rb
CHANGED
@@ -20,13 +20,13 @@ module DNN
|
|
20
20
|
|
21
21
|
# Forward propagation.
|
22
22
|
# Classes that inherit from this class must implement this method.
|
23
|
-
def forward
|
23
|
+
def forward(x)
|
24
24
|
raise NotImplementedError.new("Class '#{self.class.name}' has implement method 'forward'")
|
25
25
|
end
|
26
26
|
|
27
27
|
# Backward propagation.
|
28
28
|
# Classes that inherit from this class must implement this method.
|
29
|
-
def backward
|
29
|
+
def backward(dout)
|
30
30
|
raise NotImplementedError.new("Class '#{self.class.name}' has implement method 'update'")
|
31
31
|
end
|
32
32
|
|
@@ -181,8 +181,8 @@ module DNN
|
|
181
181
|
end
|
182
182
|
|
183
183
|
def initialize(num_nodes,
|
184
|
-
weight_initializer:
|
185
|
-
bias_initializer:
|
184
|
+
weight_initializer: Initializers::RandomNormal.new,
|
185
|
+
bias_initializer: Initializers::Zeros.new,
|
186
186
|
l1_lambda: 0,
|
187
187
|
l2_lambda: 0)
|
188
188
|
super(weight_initializer: weight_initializer, bias_initializer: bias_initializer,
|
@@ -328,16 +328,6 @@ module DNN
|
|
328
328
|
def initialize(momentum: 0.9)
|
329
329
|
super()
|
330
330
|
@momentum = momentum
|
331
|
-
@params[:gamma] = @gamma = LearningParam.new
|
332
|
-
@params[:beta] = @beta = LearningParam.new
|
333
|
-
@params[:running_mean] = nil
|
334
|
-
@params[:running_var] = nil
|
335
|
-
end
|
336
|
-
|
337
|
-
def build(model)
|
338
|
-
super
|
339
|
-
@params[:running_mean] ||= Xumo::SFloat.zeros(*shape)
|
340
|
-
@params[:running_var] ||= Xumo::SFloat.zeros(*shape)
|
341
331
|
end
|
342
332
|
|
343
333
|
def forward(x)
|
@@ -377,8 +367,12 @@ module DNN
|
|
377
367
|
private
|
378
368
|
|
379
369
|
def init_params
|
370
|
+
@params[:gamma] = @gamma = LearningParam.new
|
371
|
+
@params[:beta] = @beta = LearningParam.new
|
380
372
|
@gamma.data = Xumo::SFloat.ones(*shape)
|
381
373
|
@beta.data = Xumo::SFloat.zeros(*shape)
|
374
|
+
@params[:running_mean] = Xumo::SFloat.zeros(*shape)
|
375
|
+
@params[:running_var] = Xumo::SFloat.zeros(*shape)
|
382
376
|
end
|
383
377
|
end
|
384
378
|
end
|
data/lib/dnn/core/rnn_layers.rb
CHANGED
@@ -14,8 +14,8 @@ module DNN
|
|
14
14
|
def initialize(num_nodes,
|
15
15
|
stateful: false,
|
16
16
|
return_sequences: true,
|
17
|
-
weight_initializer:
|
18
|
-
bias_initializer:
|
17
|
+
weight_initializer: Initializers::RandomNormal.new,
|
18
|
+
bias_initializer: Initializers::Zeros.new,
|
19
19
|
l1_lambda: 0,
|
20
20
|
l2_lambda: 0)
|
21
21
|
super(weight_initializer: weight_initializer, bias_initializer: bias_initializer,
|
@@ -430,7 +430,7 @@ module DNN
|
|
430
430
|
stateful: false,
|
431
431
|
return_sequences: true,
|
432
432
|
weight_initializer: Initializers::RandomNormal.new,
|
433
|
-
bias_initializer: Initializers::Zeros.new
|
433
|
+
bias_initializer: Initializers::Zeros.new,
|
434
434
|
l1_lambda: 0,
|
435
435
|
l2_lambda: 0)
|
436
436
|
super
|
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.8.
|
4
|
+
version: 0.8.3
|
5
5
|
platform: ruby
|
6
6
|
authors:
|
7
7
|
- unagiootoro
|
8
8
|
autorequire:
|
9
9
|
bindir: exe
|
10
10
|
cert_chain: []
|
11
|
-
date: 2019-02-
|
11
|
+
date: 2019-02-18 00:00:00.000000000 Z
|
12
12
|
dependencies:
|
13
13
|
- !ruby/object:Gem::Dependency
|
14
14
|
name: numo-narray
|