tf.keras.metrics.binary_focal_crossentropy

Вычисляет потерю бинарной фокальной перекрёстной энтропии.

tf.keras.metrics.binary_focal_crossentropy(
    y_true, y_pred, gamma=2.0, from_logits=False, label_smoothing=0.0, axis=-1
)

Согласно Lin et al., 2018, это помогает применять фокальный множитель для уменьшения лёгких примеров и большего внимания на сложных примерах. По умолчанию тензор фокуса вычисляется следующим образом:

focal_factor = (1 - output)**gamma для класса 1 focal_factor = output**gamma для класса 0, где gamma — параметр фокусировки. Когда gamma = 0, эта функция эквивалентна функции потерь бинарной перекрёстной энтропии.

Использование в автономном режиме:

y_true = [[0, 1], [0, 0]]
y_pred = [[0.6, 0.4], [0.4, 0.6]]
loss = tf.keras.losses.binary_focal_crossentropy(y_true, y_pred, gamma=2)
assert loss.shape == (2,)
loss.numpy()
array([0.330, 0.206], dtype=float32)