bamtercelboo / pytorch_text_classification

text classification with my own architecture

pytorch_text_classification

• A classification task implement in pytorch for my own architecture.

Requirement

pyorch : 1.0.1
python : 3.6
cuda : 8.0/9.0 (support cuda speed up, can chose)

Usage

modify the config file, see the Config directory(here) for detail.

1、python main.py
2、python main.py --config_file ./Config/config.cfg --device cuda:0 --train -p

Model

• CNN
• BiLSTM
• Updating

Data

• SST-Binary

Result

The following test set accuracy are based on the best dev set accuracy.

Data/Model	% SST-Binary
CNN	84.23
Bi-LSTM	86.27

Reference

• cnn-lstm-bilstm-deepcnn-clstm-in-pytorch
• 基于pytorch的CNN-LSTM神经网络模型调参小结
• Convolutional Neural Networks for Sentence Classification
• Context-Sensitive Lexicon Features for Neural Sentiment Analysis

Question

• if you have any question, you can open a issue or email bamtercelboo@{gmail.com, 163.com}.

• if you have any good suggestions, you can PR or email me.