Как создать настраиваемую функцию стоимости в Keras, используя дополнительный аргумент, зависящий от изображения?

Question

Я пытаюсь создать пользовательскую функцию стоимости в Keras, которая использует дополнительный параметр, который изменяется с каждым изображением. Для каждого изображения, на котором тренируется модель, есть связанный параметр, называемый альфа, который связан с тем, насколько важно это конкретное изображение. Если есть 100 изображений, есть 100 альфа-значений, и порядок имеет значение. Как бы я реализовал это в пользовательской функции стоимости MSE?

Answer 1

tf.keras.Model.fit имеет параметр sample_weight (https://www.tensorflow.org/api_docs/python/tf/keras/Model#fit), это может помочь. Я предполагаю, что это также существует в родном Keras (хотя понятия не имею).