nn 2.1.0 → 2.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8e26103663135846646308af8fa48851d0f56d34122bfdb6172e1a8bd4675bdf
-  data.tar.gz: cffd4270bae567540673a1ec327ea416b14c3794172b7db3c589c0053bf15466
+  metadata.gz: a22a2db0abf0ff95fda30fcfdc92ec12847bd033b544cdb87ce007f69cb9f760
+  data.tar.gz: 91528c11fa35c99e0a3b0d84b863266a0107e810d8493d0489f6c7c372483fbe
 SHA512:
-  metadata.gz: abaa7793eb7e44cd44d9f49f63d7cd252f01e11e4c41d7e709cf6c8124b923d300ff6c2e2b07f410f4dbeaaa8f53a82f876e22582681ec4178364110128726ab
-  data.tar.gz: c8f830238afd0a767f5cc34a7be600c2fc481c591f8f8f433c13d0ad9647011e49f3b0ff4b87d5784a0c8ce2c01552445dc1540e252321cd856eea23c4a6f84d
+  metadata.gz: d6cf94cf86c9c0e160f369466a146b18b5114bafff755ed6ff4cb88a98898727fb67a94ce04a117b3d90761028be57777e7fb7effdca01a023586ba6a84cd1f1
+  data.tar.gz: 27cf471cece4749dea22e2a470d5e342fd33de689ff38191cfc0445a6df3655b0da5f91552766fbd2db10e0ba3bd7a17997a863a2811db2b937bddaea5dc5a60

data/document.txt

@@ -152,3 +152,4 @@ http://d.hatena.ne.jp/n_shuyo/20090913/mnist
 2018/5/16 バージョン2.0公開
 2018/6/10 バージョン2.0.1公開
 2018/6/10 バージョン2.1.0公開
+2018/6/10 バージョン2.2.0公開

data/lib/nn.rb

@@ -2,7 +2,7 @@ require "numo/narray"
 require "json"
 
 class NN
-  VERSION = "2.1"
+  VERSION = "2.2"
 
   include Numo
 
@@ -297,13 +297,12 @@ class NN::Affine
   end
 
   def backward(dout)
-    x = @x.reshape(*@x.shape, 1)
-    @d_weight = x.dot(dout.reshape(dout.shape[0], 1, dout.shape[1])).mean(0)
+    @d_weight = @x.transpose.dot(dout)
     if @nn.weight_decay > 0
       dridge = @nn.weight_decay * @nn.weights[@index]
       @d_weight += dridge
     end
-    @d_bias = dout.mean
+    @d_bias = dout.sum(0)
     dout.dot(@nn.weights[@index].transpose)
   end
 end
@@ -347,7 +346,7 @@ class NN::Identity
   end
 
   def backward(y)
-    @out - y
+    (@out - y) / @nn.batch_size
   end
 
   def loss(y)
@@ -369,7 +368,7 @@ class NN::Softmax
   end
 
   def backward(y)
-    @out - y
+    (@out - y) / @nn.batch_size
   end
 
   def loss(y)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: nn
 version: !ruby/object:Gem::Version
-  version: 2.1.0
+  version: 2.2.0
 platform: ruby
 authors:
 - unagiootoro
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2018-06-10 00:00:00.000000000 Z
+date: 2018-06-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray