Skip to content

Latest commit

 

History

History
28 lines (16 loc) · 819 Bytes

README.md

File metadata and controls

28 lines (16 loc) · 819 Bytes

Deep Pyramid Convolutional Neural Networks for Text Categorization

This is a simple version of the paper Deep Pyramid Convolutional Neural Networks for Text Categorization.

'model'

You should rewrite the Dataset class in the data/dataset.py
and put your data in '/data/train' or any other directory.

run by

python main.py --lr=0.001 --epoch=20 --batch_size=64 --gpu=0 --seed=0 --label_num=2			

Evaluation

I run the model in a dataset about AD identify.
And make a comparition between the TextCNN, LSTM and our DPCNN.

Loss of TextCNN and LSTM.

Loss of DPCNN.