Class: DNN::Initializers::RandomUniform

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

Instance Attribute Summary collapse

Instance Method Summary collapse

Methods inherited from Initializer

from_hash

Constructor Details

#initialize(min = -0.05,, max = 0.05, seed: true) ⇒ RandomUniform

Returns a new instance of RandomUniform.

Parameters:

  • min (Float) (defaults to: -0.05,)

    Min of initialization value.

  • max (Float) (defaults to: 0.05)

    Max of initialization value.



98
99
100
101
102
# File 'lib/dnn/core/initializers.rb', line 98

def initialize(min = -0.05, max = 0.05, seed: true)
  super(seed: seed)
  @min = min
  @max = max
end

Instance Attribute Details

#maxObject (readonly)

Returns the value of attribute max.



94
95
96
# File 'lib/dnn/core/initializers.rb', line 94

def max
  @max
end

#minObject (readonly)

Returns the value of attribute min.



93
94
95
# File 'lib/dnn/core/initializers.rb', line 93

def min
  @min
end

Instance Method Details

#init_param(layer, param) ⇒ Object



104
105
106
107
# File 'lib/dnn/core/initializers.rb', line 104

def init_param(layer, param)
  Xumo::SFloat.srand(@seed)
  param.data = param.data.rand(@min, @max)
end

#load_hash(hash) ⇒ Object



113
114
115
# File 'lib/dnn/core/initializers.rb', line 113

def load_hash(hash)
  initialize(hash[:min], hash[:max], seed: hash[:seed])
end

#to_hashObject



109
110
111
# File 'lib/dnn/core/initializers.rb', line 109

def to_hash
  super(min: @min, max: @max)
end