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