/MultilabelCrossEntropyLoss-Pytorch

multilabel categorical crossentropy

Primary LanguagePython

MultilabelCrossEntropyLoss-Pytorch

multilabel categorical crossentropy

This is a Pytorch implementation of multilabel crossentropy loss, which is modified from Keras version here:

苏剑林. (2020, Apr 25). 《将“softmax+交叉熵”推广到多标签分类问题 》[Blog post]. Retrieved from https://kexue.fm/archives/7359