BCELoss cu greutăți pentru etichete (cum ar fi weighted_cross_entropy_with_logits în TF) · Numărul # 5660 ·

Comentarii

Copiați linkul Citat răspuns

velikodniy comentat 9 martie 2018 •

Am implementat un analog al weighted_cross_entropy_with_logits în proiectul meu actual. Este util pentru lucrul cu seturi de date dezechilibrate. Vreau să-l adaug la PyTorch, dar mă îndoiesc dacă este cu adevărat necesar pentru alții.

De exemplu, implementarea mea:

(Desigur, testele și WeightedBCELoss trebuie scrise și ele.)

Textul a fost actualizat cu succes, dar s-au întâlnit aceste erori:

velikodniy comentat 9 martie 2018

O altă modalitate este de a adăuga un argument pos_weight la pierderea BCELoss și BCEWithLogits.

---