Максимизировать функцию в задаче бинарной классификации в керасе - PullRequest
0 голосов
/ 27 сентября 2019

Это мой первый раз здесь.Я не знаю, была ли моя грудь чистой.

Я пытаюсь применить глубокое обучение (FCN) для задачи бинарной классификации с высоким дисбалансом класса (0: 48887, 1:8862).

Я хочу развернуть следующую функцию: (#true_positive*165 - #false_positive*203) / #cases to predict.

Для дисбаланса класса я попытался class_weight, и я думаю, что для решения этой проблемы будет изменение функции потерь, но я не уверен в этом.Я буду очень признателен за любые указания о том, как я могу решить эту задачу.Спасибо!

1 Ответ

0 голосов
/ 27 сентября 2019

Используйте это как потерю:

def customLoss(true, pred):
    true_positive = true * pred
    false_positive = (1-true) * pred

    #returning negative because Keras will minimize this instead of maximize
    return - (true_positive - false_negative)

Среднее значение выборок (деление на случаи для прогнозирования) автоматически позднее.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...