Class: DNN::Losses::HuberLoss

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

Instance Method Summary collapse

Methods inherited from Loss

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

Instance Method Details

#backward(y, t) ⇒ Object 



110
111
112
113
114
115
116
117
 
# File 'lib/dnn/core/losses.rb', line 110

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 



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

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