Class: DNN::Losses::HuberLoss

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

Instance Method Summary collapse

Methods inherited from Loss

#d_regularize, #regularize, #to_hash

Instance Method Details

#backward(y) ⇒ Object 



68
69
70
71
72
73
74
75
 
# File 'lib/dnn/core/losses.rb', line 68

def backward(y)
  dout = @out - y
  if @loss > 1
    dout[dout >= 0] = 1
    dout[dout < 0] = -1
  end
  dout
end

#forward(out, y, layers) ⇒ Object 



60
61
62
63
64
65
66
 
# File 'lib/dnn/core/losses.rb', line 60

def forward(out, y, layers)
  @out = out
  loss = loss_l1(y)
  loss = loss > 1 ? loss : loss_l2(y)
  #@loss = loss + regularize(layers)

  @loss = loss
end