BCELoss з вагами для міток (наприклад, зважений_крос_ентропій_з_логітами в TF) · Випуск # 5660 ·

Коментарі

Копіювати посилання Цитувати відповідь

bceloss

великодній прокоментував 9 березня 2018 р. •

У своєму поточному проекті я реалізував аналог зважених_кросів_ентропій_ з_логітами. Це корисно для роботи з незбалансованими наборами даних. Я хочу додати його до PyTorch, але я сумніваюся, чи він справді потрібен іншим.

Наприклад, моя реалізація:

(Звичайно, тести та WeightedBCELoss також слід писати.)

Текст успішно оновлено, але виявлені такі помилки:

великодній прокоментував 9 березня 2018 р

Інший спосіб - додати аргумент pos_weight до втрати BCELoss та BCEWithLogits.