Class: DNN::Losses::SigmoidCrossEntropy

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

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Methods inherited from Loss

#backward, #forward

Constructor Details

#initialize(eps: 1e-7) ⇒ SigmoidCrossEntropy

Returns a new instance of SigmoidCrossEntropy.

Parameters:

  • eps (Float) (defaults to: 1e-7) —

    Value to avoid nan.



153
154
155
 
# File 'lib/dnn/core/losses.rb', line 153

def initialize(eps: 1e-7)
  @eps = eps
end

Instance Attribute Details

#eps ⇒ Float

Return the eps value.

Returns:

  • (Float) —

    Return the eps value.



146
147
148
 
# File 'lib/dnn/core/losses.rb', line 146

def eps
  @eps
end

Class Method Details

.from_hash(hash) ⇒ Object 



148
149
150
 
# File 'lib/dnn/core/losses.rb', line 148

def self.from_hash(hash)
  SigmoidCrossEntropy.new(eps: hash[:eps])
end

Instance Method Details

#to_hash ⇒ Object 



157
158
159
 
# File 'lib/dnn/core/losses.rb', line 157

def to_hash
  super(eps: @eps)
end