Class: DNN::Losses::HuberLoss

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

Instance Method Summary collapse

Methods inherited from Loss

from_hash, #load_hash, #loss, #regularizers_backward, #regularizers_forward, #to_hash

Instance Method Details

#backward(y, t) ⇒ Object 



102
103
104
105
106
107
108
109
 
# File 'lib/dnn/core/losses.rb', line 102

def backward(y, t)
  dy = y - t
  if @loss_value > 1
    dy[dy >= 0] = 1
    dy[dy < 0] = -1
  end
  dy
end

#forward(y, t) ⇒ Object 



97
98
99
100
 
# File 'lib/dnn/core/losses.rb', line 97

def forward(y, t)
  loss_l1_value = loss_l1(y, t)
  @loss_value = loss_l1_value > 1 ? loss_l1_value : loss_l2(y, t)
end