Условные потери на основе второго выхода

Question

У меня есть (составная) модель, которая прогнозирует 2 выхода. Для output1 у меня фиксированная потеря binary_crossentropy. Для output2 мне нужна потеря, которая выглядит так:

if output1 > 0.8:
    return binary_crossentropy(output2, output2_true)
else:
    return 0

Как я могу реализовать эту логику для потери в кератах?

Answer 1

Для этой цели, я думаю, вам нужно написать собственную функцию потерь и объединить 2 NN с помощью Concatenate ().Ваша функция условных потерь может быть затем оценена и распространена за 1 проход на основе объединенных результатов 1 и 2. Может быть что-то вроде: https://github.com/keras-team/keras/issues/2662