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



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

def initialize(learning_rate)
  @learning_rate = learning_rate
end

Instance Attribute Details

#learning_rateObject

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

#update(layer) ⇒ Object

Update layer has params.



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

def update(layer) end