- A classification task implement in pytorch for my own architecture.
pyorch : 1.0.1
python : 3.6
cuda : 8.0/9.0 (support cuda speed up, can chose)
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
- CNN
- BiLSTM
- Updating
- SST-Binary
The following test set accuracy are based on the best dev set accuracy.
Data/Model | % SST-Binary |
---|---|
CNN | 84.23 |
Bi-LSTM | 86.27 |
- cnn-lstm-bilstm-deepcnn-clstm-in-pytorch
- 基于pytorch的CNN-LSTM神经网络模型调参小结
- Convolutional Neural Networks for Sentence Classification
- Context-Sensitive Lexicon Features for Neural Sentiment Analysis
-
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.