ruby-dnn 0.10.0 → 0.10.1
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/model.rb +1 -1
- data/lib/dnn/version.rb +1 -1
- metadata +1 -1
checksums.yaml
CHANGED
@@ -1,7 +1,7 @@
|
|
1
1
|
---
|
2
2
|
SHA256:
|
3
|
-
metadata.gz:
|
4
|
-
data.tar.gz:
|
3
|
+
metadata.gz: 2283358c50e7c60f4f1d67763f16498544a370158c27dd9df4865b8e9e75f2c4
|
4
|
+
data.tar.gz: 1d6a2fb9043a7d652906cfd10b8c2301eeb4d110687aef96a62e6ef26e01fed9
|
5
5
|
SHA512:
|
6
|
-
metadata.gz:
|
7
|
-
data.tar.gz:
|
6
|
+
metadata.gz: 3827f751e0f7b2ff4f3df5a88d4b768ee986ec4036ec6a3324de52504cbe1324c55c8b6bddfe8f84f4e39d783ed3bfd93b65a8c45947401bf00fb3f34810ac2e
|
7
|
+
data.tar.gz: 2292999ec27ba89d23d0f592dc314ee8350caaff164eee8faf5e2e353e1415407af9a793a78c4dfa77aea13f9ed100cb6b25a69a49e93ffcd1dfffe2d28fe4d2
|
data/lib/dnn/core/model.rb
CHANGED
@@ -285,7 +285,7 @@ module DNN
|
|
285
285
|
sum_loss = 0
|
286
286
|
(x.shape[0].to_f / batch_size).ceil.times do |i|
|
287
287
|
x_batch, y_batch = dataset.next_batch(batch_size)
|
288
|
-
x_batch, y_batch = before_batch_cbk.call(x_batch, y_batch,
|
288
|
+
x_batch, y_batch = before_batch_cbk.call(x_batch, y_batch, false) if before_batch_cbk
|
289
289
|
x_batch = forward(x_batch, false)
|
290
290
|
sigmoid = Sigmoid.new
|
291
291
|
batch_size.times do |j|
|
data/lib/dnn/version.rb
CHANGED