├── bert_multilabel_cls.py ├── data └── train.json ├── data_helper.py ├── data_preprocess.py ├── predict.py ├── readme.md ├── requirements.txt └── train.py /bert_multilabel_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/murray-z/multi_label_classification/HEAD/bert_multilabel_cls.py -------------------------------------------------------------------------------- /data/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/murray-z/multi_label_classification/HEAD/data/train.json -------------------------------------------------------------------------------- /data_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/murray-z/multi_label_classification/HEAD/data_helper.py -------------------------------------------------------------------------------- /data_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/murray-z/multi_label_classification/HEAD/data_preprocess.py -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/murray-z/multi_label_classification/HEAD/predict.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/murray-z/multi_label_classification/HEAD/readme.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | transformers=4.18.0 -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/murray-z/multi_label_classification/HEAD/train.py --------------------------------------------------------------------------------