Class: DNN::Optimizers::Optimizer

Inherits:
Object
  • Object
show all
Defined in:
lib/dnn/core/optimizers.rb

Overview

Super class of all optimizer classes.

Direct Known Subclasses

Adam, RMSProp, SGD

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(learning_rate) ⇒ Optimizer

Returns a new instance of Optimizer.



8
9
10
 
# File 'lib/dnn/core/optimizers.rb', line 8

def initialize(learning_rate)
  @learning_rate = learning_rate
end

Instance Attribute Details

#learning_rate ⇒ Object

Returns the value of attribute learning_rate.



6
7
8
 
# File 'lib/dnn/core/optimizers.rb', line 6

def learning_rate
  @learning_rate
end

Instance Method Details

#to_hash(hash) ⇒ Object 



15
16
17
 
# File 'lib/dnn/core/optimizers.rb', line 15

def to_hash(hash)
  {name: self.class.name, learning_rate: @learning_rate}.merge(hash)
end

#update(layer) ⇒ Object

Update layer has params.



13
 
# File 'lib/dnn/core/optimizers.rb', line 13

def update(layer) end