Веса в BCEWithLogitsLoss - PullRequest
       100

Веса в BCEWithLogitsLoss

1 голос
/ 30 апреля 2020

У меня проблема с многоклассовой классификационной задачей. У меня есть набор данных размером 33000, каждый из которых содержит 104 класса. Я разделил набор данных на 16500 выборок с метками, такими как [1, 0, 1, 0, 0,…], [0, 1, 1, 0, 1,…], [1, 0, 0, 0] (каждый метка содержит как минимум один элемент 1) и 16500 меток, таких как [0, 0, 0,…], [0, 0, 0,…] (все элементы во всех метках равны 0). При расчете pos_count для каждого класса число pos_count_0 для класса 0 - это то, сколько из 1 появляется в первой позиции каждой метки в моем наборе данных. Для класса 1 pos_count_1 число 1 во второй позиции и так далее. И после этого pos_weight класса 0 будет (33000-pos_count_0) / pos_count_0, pos_weight класса 1 будет (33000-pos_count_1) / pos_count_1? Я немного запутался, как рассчитываются neg_count и pos_count для класса.

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