Skip to content

Cheneng/DPCNN

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

Deep Pyramid Convolutional Neural Networks for Text Categorization in PyTorch

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages