├── .gitignore ├── .travis.yml ├── MANIFEST.in ├── README.md ├── examples ├── test.py └── utils.py ├── requirements.txt ├── setup.py ├── tests ├── __init__.py └── simple_test.py └── text_classification ├── __init__.py ├── base.py ├── layers ├── __init__.py └── attention.py └── models ├── __init__.py ├── text_birnn.py ├── text_clstm1.py ├── text_clstm2.py ├── text_cnn.py ├── text_fasttext.py ├── text_han.py ├── text_mlp.py ├── text_rcnn.py ├── text_rnn.py └── text_textcnn.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/.gitignore -------------------------------------------------------------------------------- /.travis.yml: -------------------------------------------------------------------------------- 1 | language: python 2 | 3 | script: 4 | - coverage run setup.py test 5 | -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- 1 | recursive-include text_classification * -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/README.md -------------------------------------------------------------------------------- /examples/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/examples/test.py -------------------------------------------------------------------------------- /examples/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/examples/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | Keras==2.0.8 2 | tensorflow==1.4.1 3 | coverage==4.5 4 | -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/tests/__init__.py -------------------------------------------------------------------------------- /tests/simple_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/tests/simple_test.py -------------------------------------------------------------------------------- /text_classification/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/__init__.py -------------------------------------------------------------------------------- /text_classification/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/base.py -------------------------------------------------------------------------------- /text_classification/layers/__init__.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | 3 | from attention import AttLayer 4 | -------------------------------------------------------------------------------- /text_classification/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/layers/attention.py -------------------------------------------------------------------------------- /text_classification/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/__init__.py -------------------------------------------------------------------------------- /text_classification/models/text_birnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_birnn.py -------------------------------------------------------------------------------- /text_classification/models/text_clstm1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_clstm1.py -------------------------------------------------------------------------------- /text_classification/models/text_clstm2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_clstm2.py -------------------------------------------------------------------------------- /text_classification/models/text_cnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_cnn.py -------------------------------------------------------------------------------- /text_classification/models/text_fasttext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_fasttext.py -------------------------------------------------------------------------------- /text_classification/models/text_han.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_han.py -------------------------------------------------------------------------------- /text_classification/models/text_mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_mlp.py -------------------------------------------------------------------------------- /text_classification/models/text_rcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_rcnn.py -------------------------------------------------------------------------------- /text_classification/models/text_rnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_rnn.py -------------------------------------------------------------------------------- /text_classification/models/text_textcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qiangsiwei/text_classification/HEAD/text_classification/models/text_textcnn.py --------------------------------------------------------------------------------