├── LICENSE.txt ├── README.md ├── config ├── grid_search_cnn.ini └── imdb.ini ├── dataHelper.py ├── dataloader ├── Dataset.py ├── __init__.py ├── ag.py ├── glove.py ├── imdb.py ├── mr.py ├── sst.py └── torch_text_demo │ ├── imdb.py │ ├── sst.py │ └── trec.py ├── docs ├── data_config.md ├── data_config_en.md ├── windows_torch.md └── windows_torch_en.md ├── main.py ├── models ├── BERTFast.py ├── BaseModel.py ├── BiBloSA.py ├── CNN.py ├── CNNBasic.py ├── CNNInception.py ├── CNNKim.py ├── CNNMultiLayer.py ├── CNNText.py ├── CNN_Inception.py ├── Capsule.py ├── ConvS2S.py ├── DiSAN.py ├── FastText.py ├── GPTModel.py ├── LSTM.py ├── LSTMBI.py ├── LSTMStack.py ├── LSTMTree.py ├── LSTMwithAttention.py ├── MLP.py ├── MemoryNetwork.py ├── QuantumCNN.py ├── RCNN.py ├── RNN_CNN.py ├── SelfAttention.py ├── Transformer.py ├── XLNetTransformer.py ├── __init__.py └── ensemble_strategy.py ├── opts.py ├── push.bash ├── search.sh ├── trandition.py └── utils.py /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/README.md -------------------------------------------------------------------------------- /config/grid_search_cnn.ini: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/config/grid_search_cnn.ini -------------------------------------------------------------------------------- /config/imdb.ini: -------------------------------------------------------------------------------- 1 | [COMMON] 2 | dataset = imdb;sst 3 | 4 | -------------------------------------------------------------------------------- /dataHelper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataHelper.py -------------------------------------------------------------------------------- /dataloader/Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/Dataset.py -------------------------------------------------------------------------------- /dataloader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/__init__.py -------------------------------------------------------------------------------- /dataloader/ag.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/ag.py -------------------------------------------------------------------------------- /dataloader/glove.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/glove.py -------------------------------------------------------------------------------- /dataloader/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/imdb.py -------------------------------------------------------------------------------- /dataloader/mr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/mr.py -------------------------------------------------------------------------------- /dataloader/sst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/sst.py -------------------------------------------------------------------------------- /dataloader/torch_text_demo/imdb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/torch_text_demo/imdb.py -------------------------------------------------------------------------------- /dataloader/torch_text_demo/sst.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/torch_text_demo/sst.py -------------------------------------------------------------------------------- /dataloader/torch_text_demo/trec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/dataloader/torch_text_demo/trec.py -------------------------------------------------------------------------------- /docs/data_config.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/docs/data_config.md -------------------------------------------------------------------------------- /docs/data_config_en.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/docs/data_config_en.md -------------------------------------------------------------------------------- /docs/windows_torch.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/docs/windows_torch.md -------------------------------------------------------------------------------- /docs/windows_torch_en.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/docs/windows_torch_en.md -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/main.py -------------------------------------------------------------------------------- /models/BERTFast.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/BERTFast.py -------------------------------------------------------------------------------- /models/BaseModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/BaseModel.py -------------------------------------------------------------------------------- /models/BiBloSA.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/BiBloSA.py -------------------------------------------------------------------------------- /models/CNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/CNN.py -------------------------------------------------------------------------------- /models/CNNBasic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/CNNBasic.py -------------------------------------------------------------------------------- /models/CNNInception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/CNNInception.py -------------------------------------------------------------------------------- /models/CNNKim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/CNNKim.py -------------------------------------------------------------------------------- /models/CNNMultiLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/CNNMultiLayer.py -------------------------------------------------------------------------------- /models/CNNText.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/CNNText.py -------------------------------------------------------------------------------- /models/CNN_Inception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/CNN_Inception.py -------------------------------------------------------------------------------- /models/Capsule.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/Capsule.py -------------------------------------------------------------------------------- /models/ConvS2S.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | 3 | -------------------------------------------------------------------------------- /models/DiSAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/DiSAN.py -------------------------------------------------------------------------------- /models/FastText.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/FastText.py -------------------------------------------------------------------------------- /models/GPTModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/GPTModel.py -------------------------------------------------------------------------------- /models/LSTM.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/LSTM.py -------------------------------------------------------------------------------- /models/LSTMBI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/LSTMBI.py -------------------------------------------------------------------------------- /models/LSTMStack.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/LSTMTree.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | 3 | # https://github.com/dasguptar/treelstm.pytorch -------------------------------------------------------------------------------- /models/LSTMwithAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/LSTMwithAttention.py -------------------------------------------------------------------------------- /models/MLP.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/MLP.py -------------------------------------------------------------------------------- /models/MemoryNetwork.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/MemoryNetwork.py -------------------------------------------------------------------------------- /models/QuantumCNN.py: -------------------------------------------------------------------------------- 1 | # -*- coding: utf-8 -*- 2 | 3 | -------------------------------------------------------------------------------- /models/RCNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/RCNN.py -------------------------------------------------------------------------------- /models/RNN_CNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/RNN_CNN.py -------------------------------------------------------------------------------- /models/SelfAttention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/SelfAttention.py -------------------------------------------------------------------------------- /models/Transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/Transformer.py -------------------------------------------------------------------------------- /models/XLNetTransformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/XLNetTransformer.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/ensemble_strategy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/models/ensemble_strategy.py -------------------------------------------------------------------------------- /opts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/opts.py -------------------------------------------------------------------------------- /push.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/push.bash -------------------------------------------------------------------------------- /search.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/search.sh -------------------------------------------------------------------------------- /trandition.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/trandition.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FreedomIntelligence/TextClassificationBenchmark/HEAD/utils.py --------------------------------------------------------------------------------