Skip to content

PyTorch Implementation of Asymetric Cross Entropy Loss (Loss Correction for PU Learning)

License

Notifications You must be signed in to change notification settings

carlomarxdk/AsymmetricLoss

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Asymmetric Cross Entropy Loss

Implementation of the Asymmetric Cross Entropy Loss for the Positive-Unlabeled Learning (PU-Learning) for PyTorch. The method is adapted from the Asymmetric Loss for Positive-Unlabeled Learning paper [1] - it is not one-to-one implementation.

References

[1] Wang, C., Pu, J., Xu, Z. and Zhang, J., 2021, July. Asymmetric Loss for Positive-Unlabeled Learning. In 2021 IEEE International Conference on Multimedia and Expo (ICME) (pp. 1-6). IEEE.

About

PyTorch Implementation of Asymetric Cross Entropy Loss (Loss Correction for PU Learning)

Topics

Resources

License

Stars

Watchers

Forks