class DNN::Losses::Hinge

Public Instance Methods

backward_node(d) click to toggle source
# File lib/dnn/core/losses.rb, line 104
def backward_node(d)
  a = Xumo::SFloat.ones(*@a.shape)
  a[@a <= 0] = 0
  d * (a * -@t) / a.shape[0]
end
forward_node(y, t) click to toggle source
# File lib/dnn/core/losses.rb, line 98
def forward_node(y, t)
  @t = t
  @a = 1 - y * t
  Xumo::SFloat.maximum(0, @a).mean(0).sum
end