當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文

Python tf.keras.losses.huber用法及代碼示例


計算 Huber 損失值。

用法

tf.keras.losses.huber(
    y_true, y_pred, delta=1.0
)

參數

  • y_true 真實目標的張量。
  • y_pred 預測目標的張量。
  • delta 浮點數，Huber 損失函數從二次變為線性的點。

返回

  • 每個樣本具有一個標量損失條目的張量。

對於 error = y_true - y_pred 中的每個值 x：

loss = 0.5 * x^2                  if |x| <= d
loss = d * |x| - 0.5 * d^2        if |x| > d

其中 d 是delta.看：https://en.wikipedia.org/wiki/Huber_loss

相關用法


注：本文由純淨天空篩選整理自tensorflow.org大神的英文原創作品 tf.keras.losses.huber。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。