Блог

Главная - Вопросы по программированию - Как использовать веса выборки в пользовательской функции потерь в Keras?

Как использовать веса выборки в пользовательской функции потерь в Keras?

#keras #deep-learning

#keras #глубокое обучение

Вопрос:

Я использую пользовательскую функцию потерь в keras . Теперь я хочу использовать веса выборки в Keras .

Я искал в Google, и в какой-то статье предлагается model.fit(X,y,sample_weight= custom_weights)

Но я хочу использовать вес выборки непосредственно в пользовательской функции потерь. Моя пользовательская функция потерь довольно сложная, и по какой-то причине мне нужно обрабатывать вес выборки напрямую.

например:

 custom_weights = np.array([1,2,3,4,5,6,7,8,9,10])

#my failed attempt
def custom_loss_function(y_true, y_pred , custom_weights):
    return K.mean(K.abs(y_pred - y_true) * custom_weights), axis=-1)

примечание: моя реальная функция custom_loss_function очень сложная. В этом вопросе я использую «MAE» в качестве примера, чтобы упростить проблему, чтобы мы могли сосредоточиться на ответе «как использовать веса выборки в custom_loss_function «

как правильно выполнить эту задачу?