Method: Eps::Metrics.log_loss

Defined in:
lib/eps/metrics.rb

.log_loss(y_true, y_pred, eps: 1e-15, weight: nil) ⇒ Object



33
34
35
36
37
# File 'lib/eps/metrics.rb', line 33

def log_loss(y_true, y_pred, eps: 1e-15, weight: nil)
  check_size(y_true, y_pred)
  p = y_pred.map { |yp| yp.clamp(eps, 1 - eps) }
  mean(y_true.zip(p).map { |yt, pi| yt == 1 ? -Math.log(pi) : -Math.log(1 - pi) }, weight: weight)
end