Class: DNN::Initializers::RandomNormal

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

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

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

Returns a new instance of RandomNormal.



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

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

Instance Attribute Details

#meanObject (readonly)

Returns the value of attribute mean.



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

def mean
  @mean
end

#stdObject (readonly)

Returns the value of attribute std.



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

def std
  @std
end

Class Method Details

.load_hash(hash) ⇒ Object



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

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

Instance Method Details

#init_param(layer, param) ⇒ Object



38
39
40
# File 'lib/dnn/core/initializers.rb', line 38

def init_param(layer, param)
  param.data = param.data.rand_norm(@mean, @std)
end

#to_hashObject



42
43
44
# File 'lib/dnn/core/initializers.rb', line 42

def to_hash
  super({mean: @mean, std: @std})
end