Class: DNN::Initializers::RandomNormal

Inherits:
Initializer show all
Defined in:
lib/dnn/core/initializers.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(mean = 0, std = 0.05) ⇒ RandomNormal

Returns a new instance of RandomNormal.



27
28
29
30
# File 'lib/dnn/core/initializers.rb', line 27

def initialize(mean = 0, std = 0.05)
  @mean = mean
  @std = std
end

Class Method Details

.load_hash(hash) ⇒ Object



23
24
25
# File 'lib/dnn/core/initializers.rb', line 23

def self.load_hash(hash)
  self.new(hash[:mean], hash[:std])
end

Instance Method Details

#init_param(layer, param_key) ⇒ Object



32
33
34
# File 'lib/dnn/core/initializers.rb', line 32

def init_param(layer, param_key)
  super(layer, param_key, layer.params[param_key].rand_norm(@mean, @std))
end

#to_hashObject



36
37
38
# File 'lib/dnn/core/initializers.rb', line 36

def to_hash
  {name: self.class.name, mean: @mean, std: @std}
end