ops.py 文件源码

def huber_loss(labels, predictions, delta=1.0):
    ''' Huber loss: L2 befor delta, L1 after delta '''
    residual = tf.abs(predictions - labels)
    condition = tf.less(residual, delta)
    small_res = 0.5 * tf.square(residual)
    large_res = delta * residual - 0.5 * tf.square(delta)
    return tf.where(condition, small_res, large_res)