├── .vscode └── settings.json ├── LICENSE ├── README.md ├── dataset.py ├── figure └── model.png ├── model.py ├── predict.py ├── requirements.txt └── train.py /.vscode/settings.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/.vscode/settings.json -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/README.md -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/dataset.py -------------------------------------------------------------------------------- /figure/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/figure/model.png -------------------------------------------------------------------------------- /model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/model.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/predict.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | torch 2 | transformers 3 | tqdm 4 | numpy 5 | sklearn 6 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/illiterate/BertClassifier/HEAD/train.py --------------------------------------------------------------------------------