ruby-dnn 0.8.4 → 0.8.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 61defa3dd2c7b95ac7d3c3484789abfc7ec792be0fd4dea728fee628479fcf76
-  data.tar.gz: 12e3a398119eb5a64eec7578b6fb3263dca4a5a1028300ab97b47194766bf65b
+  metadata.gz: 3b897e7f51a4bf649e3bc9b0207dd3dbea30fc35d50ca4cf5ff08c7f8daacb4c
+  data.tar.gz: '078b1fa114f3abe04cc40e2e01211f3a563c2796bc9da78b904478ad0a5e83ac'
 SHA512:
-  metadata.gz: cf23b2ea6cb6d00fa2a6b822c309e0db6677f890be18c43f910aae41ea3fee0342734434f349c98e597620b2729d3a165aa8c135f4ecb94d53343bbe49709f9a
-  data.tar.gz: 44c73b4b066a2026a2fa5d6ca81c52fdefb4c411ac9a241290c254a5071a9aa35d3ad63c71bbe09583980523f24ed8d78988fb37656b098d6896eab267585f16
+  metadata.gz: fe348f3964752cdc520d2f571da843a26073795de1033a823757e388f506e800fd310ffe6063d3cb415d55502c5c9576162cb1404a925088dced844b143aa129
+  data.tar.gz: d3347bb4ed8c6b2ad9945252947aa44c44eda188ec64cb5aa4ac371d7f3a426241f9c76c67eb12de42854ceebf4aa796a2762b43b7395a09a75e234e403b6619

data/API-Reference.ja.md

@@ -2,7 +2,7 @@
 ruby-dnnのAPIリファレンスです。このリファレンスでは、APIを利用するうえで必要となるクラスとメソッドしか記載していません。
 そのため、プログラムの詳細が必要な場合は、ソースコードを参照してください。
 
-最終更新バージョン:0.8.2
+最終更新バージョン:0.8.5
 
 # module DNN
 ruby-dnnの名前空間をなすモジュールです。
@@ -152,7 +152,7 @@ epoch_proc
 Integer  
 損失関数の値を返します。
 
-## def accurate(x, y, batch_size = 1, &batch_proc)
+## def accurate(x, y, batch_size = 100, &batch_proc)
 学習結果をもとに認識率を返します。
 ### arguments
 * Numo::SFloat x  
@@ -216,6 +216,13 @@ layer_classで指定されたクラスのレイヤーをindexで取得します
 Layer  
 対象のレイヤーのインスタンス。
 
+## def get_all_layers
+モデルが持つ全てのレイヤー(モデルが持つ下位のモデルのレイヤーも含む)を取得します。
+### arguments
+なし。
+### return
+Array  
+モデルの持つすべてのレイヤーの配列
 
 # module Layers
 レイヤーの名前空間をなすモジュールです。

data/lib/dnn/core/layers.rb

@@ -275,7 +275,7 @@ module DNN
       end
 
       def dloss
-        @model.layers.select { |layer| layer.is_a?(Connection) }.each do |layer|
+        @model.get_all_layers.select { |layer| layer.is_a?(Connection) }.each do |layer|
           layer.dlasso
           layer.dridge
         end
@@ -284,13 +284,13 @@ module DNN
       private
 
       def lasso
-        @model.layers.select { |layer| layer.is_a?(Connection) }
-                     .reduce(0) { |sum, layer| sum + layer.lasso }
+        @model.get_all_layers.select { |layer| layer.is_a?(Connection) }
+                             .reduce(0) { |sum, layer| sum + layer.lasso }
       end
     
       def ridge
-        @model.layers.select { |layer| layer.is_a?(Connection) }
-                     .reduce(0) { |sum, layer| sum + layer.ridge }
+        @model.get_all_layers.select { |layer| layer.is_a?(Connection) }
+                             .reduce(0) { |sum, layer| sum + layer.ridge }
       end
     end
     

data/lib/dnn/core/model.rb

@@ -216,6 +216,12 @@ module DNN
         @layers.select { |layer| layer.is_a?(layer_class) }[index]
       end
     end
+
+    def get_all_layers
+      @layers.map { |layer|
+        layer.is_a?(Model) ? layer.get_all_layers : layer
+      }.flatten
+    end
   
     def forward(x, training)
       @training = training
@@ -291,7 +297,7 @@ module DNN
 
     def layers_shape_check
       @layers.each.with_index do |layer, i|
-        prev_shape = layer.prev_layer.shape
+        prev_shape = layer.is_a?(Layers::Layer) ? layer.prev_layer.shape : layer.layers[-1]
         if layer.is_a?(Layers::Dense)
           if prev_shape.length != 1
             raise DNN_ShapeError.new("layer index(#{i}) Dense:  The shape of the previous layer is #{prev_shape}. The shape of the previous layer must be 1 dimensional.")

data/lib/dnn/core/rnn_layers.rb

@@ -94,16 +94,20 @@ module DNN
         end
       end
 
-      def dlasso2
+      def dlasso
         if @l1_lambda > 0
-          dlasso = Xumo::SFloat.ones(*@weight2.data.shape)
-          dlasso[@weight2.data < 0] = -1
-          @weight2.grad += @l1_lambda * dlasso
+          dlasso = Xumo::SFloat.ones(*@weight.data.shape)
+          dlasso[@weight.data < 0] = -1
+          @weight.grad += @l1_lambda * dlasso
+          dlasso2 = Xumo::SFloat.ones(*@weight2.data.shape)
+          dlasso2[@weight2.data < 0] = -1
+          @weight2.grad += @l1_lambda * dlasso2
         end
       end
 
-      def dridge2
+      def dridge
         if @l2_lambda > 0
+          @weight.grad += l2_lambda * @weight.data
           @weight2.grad += l2_lambda * @weight2.data
         end
       end

data/lib/dnn/version.rb

@@ -1,3 +1,3 @@
 module DNN
-  VERSION = "0.8.4"
+  VERSION = "0.8.5"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby-dnn
 version: !ruby/object:Gem::Version
-  version: 0.8.4
+  version: 0.8.5
 platform: ruby
 authors:
 - unagiootoro
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2019-02-21 00:00:00.000000000 Z
+date: 2019-02-22 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: numo-narray