Class: DNN::Layers::SimpleRNN
- Inherits:
-
RNN
- Object
- Layer
- HasParamLayer
- Connection
- RNN
- DNN::Layers::SimpleRNN
- Defined in:
- lib/dnn/core/rnn_layers.rb
Instance Attribute Summary
Attributes inherited from RNN
Attributes inherited from Connection
Attributes inherited from HasParamLayer
Class Method Summary collapse
Instance Method Summary collapse
-
#initialize(num_nodes, stateful: false, return_sequences: true, activation: nil, weight_initializer: nil, bias_initializer: nil, l1_lambda: 0, l2_lambda: 0) ⇒ SimpleRNN
constructor
A new instance of SimpleRNN.
- #to_hash ⇒ Object
Methods inherited from RNN
#backward, #dlasso, #dlasso2, #dridge, #dridge2, #forward, #lasso, #reset_state, #ridge, #shape
Methods inherited from Connection
#dlasso, #dridge, #lasso, #ridge
Methods inherited from HasParamLayer
Methods inherited from Layer
#backward, #build, #built?, #forward, #prev_layer, #shape
Constructor Details
#initialize(num_nodes, stateful: false, return_sequences: true, activation: nil, weight_initializer: nil, bias_initializer: nil, l1_lambda: 0, l2_lambda: 0) ⇒ SimpleRNN
Returns a new instance of SimpleRNN.
169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 |
# File 'lib/dnn/core/rnn_layers.rb', line 169 def initialize(num_nodes, stateful: false, return_sequences: true, activation: nil, weight_initializer: nil, bias_initializer: nil, l1_lambda: 0, l2_lambda: 0) super(num_nodes, stateful: stateful, return_sequences: return_sequences, weight_initializer: weight_initializer, bias_initializer: bias_initializer, l1_lambda: 0, l2_lambda: 0) @activation = (activation || Tanh.new) end |
Class Method Details
.load_hash(hash) ⇒ Object
156 157 158 159 160 161 162 163 164 165 166 167 |
# File 'lib/dnn/core/rnn_layers.rb', line 156 def self.load_hash(hash) simple_rnn = self.new(hash[:num_nodes], stateful: hash[:stateful], return_sequences: hash[:return_sequences], activation: Util.load_hash(hash[:activation]), weight_initializer: Util.load_hash(hash[:weight_initializer]), bias_initializer: Util.load_hash(hash[:bias_initializer]), l1_lambda: hash[:l1_lambda], l2_lambda: hash[:l2_lambda]) simple_rnn.h = Xumo::SFloat.cast(hash[:h]) simple_rnn end |
Instance Method Details
#to_hash ⇒ Object
187 188 189 |
# File 'lib/dnn/core/rnn_layers.rb', line 187 def to_hash super({activation: @activation.to_hash}) end |