Class: DNN::Initializers::Initializer

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

Direct Known Subclasses

Const, He, RandomNormal, RandomUniform, Xavier, Zeros

Instance Method Summary collapse

Constructor Details

#initialize(seed = false) ⇒ Initializer

Returns a new instance of Initializer.



5
6
7
 
# File 'lib/dnn/core/initializers.rb', line 5

def initialize(seed = false)
  @seed = seed == true ? rand(1 << 31) : seed
end

Instance Method Details

#init_param(layer, param) ⇒ Object

Raises:

  • (NotImplementedError) 


9
10
11
 
# File 'lib/dnn/core/initializers.rb', line 9

def init_param(layer, param)
  raise NotImplementedError.new("Class '#{self.class.name}' has implement method 'init_params'")
end

#to_hash(merge_hash = nil) ⇒ Object 



13
14
15
16
17
 
# File 'lib/dnn/core/initializers.rb', line 13

def to_hash(merge_hash = nil)
  hash = {class: self.class.name, seed: @seed}
  hash.merge!(merge_hash) if merge_hash
  hash
end