- Notifications
You must be signed in to change notification settings - Fork0
fhaoguo/toutiao_news_cls
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
本项目方案适合新闻短文本分类，可以在此基础上进一步完善实现其他短文本分类任务，欢迎fork 和 star。数据及模型因存储空间限制请自行搜索获取。
| f1 | precision | recall | accuracy | |
|---|---|---|---|---|
| RandomForest(epoches=10) | 0.4773 | 0.4980 | 0.4693 | 0.5048 |
| TextCNN(epoches=2) | 0.5188 | 0.5395 | 0.5077 | 0.5266 |
| Bert(epoches=2) | 0.5709 | 0.5698 | 0.5762 | 0.5690 |
| BertWithHead(epoches=2) | 0.5536 | 0.5815 | 0.5514 | 0.5654 |
| NeZhaWithHead(epoches=2) | 0.5524 | 0.5774 | 0.5439 | 0.5622 |
| FocalLoss(epoches=1) | ||||
| 对抗训练FGM(epoches=1) | ||||
| 对抗训练PGD(epoches=1) |
Analysis.ipynbmost_simple_way.py详情点击text_cnn
- 详情点击bert_base
- 详情点击bert_head_base
详情点击nezha_head_base