Skip to content

HUSTHY/mutil_label_calssification_ccks_2021

Repository files navigation

A PyTorch implementation of mutil_label text classification

其中损失函数采用了sigmoid+BCE
苏神提出的——multilabel_crossentropy

将“softmax+交叉熵”推广到多标签分类问题

用法

python train_bert_mutillabel_classification.py

结果

BCE     train_acc:1.0000 val_acc:0.9720------best_acc:0.9720
MLCE    train_acc:1.0000 val_acc:0.9790------best_acc:0.9790

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages