Skip to content

dalinvip/pytorch_text_classification

Repository files navigation

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

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.