• Stars
    star
    198
  • Rank 196,898 (Top 4 %)
  • Language
    Python
  • Created over 6 years ago
  • Updated over 5 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

Deep Pyramid Convolutional Neural Networks for Text Categorization in PyTorch

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.