fastNLP/reproduction/text_classification/README.md
2019-07-11 00:03:24 +08:00

1.2 KiB
Raw Blame History

text_classification任务模型复现

这里使用fastNLP复现以下模型

char_cnn :论文链接Character-level Convolutional Networks for Text Classification

dpcnn:论文链接Deep Pyramid Convolutional Neural Networks for TextCategorization

HAN:论文链接Hierarchical Attention Networks for Document Classification

LSTM+self_attention:论文链接A Structured Self-attentive Sentence Embedding

AWD-LSTM:论文链接Regularizing and Optimizing LSTM Language Models

数据集及复现结果汇总

使用fastNLP复现的结果vs论文汇报结果(/前为fastNLP实现后面为论文报道,-表示论文没有在该数据集上列出结果)

model name yelp_p yelp_f sst-2 IMDB
char_cnn 93.80/95.12 - - -
dpcnn 95.50/97.36 - - -
HAN - - - -
LSTM 95.74/- 64.16/- - 88.52/-
AWD-LSTM 95.96/- 64.74/- - 88.91/-
LSTM+self_attention 96.34/- 65.78/- - 89.53/-