├── LICENSE ├── README.md ├── checkpoints └── readme.txt ├── data ├── test.json └── train.json ├── data_preprocessing ├── data_prepro.py └── tools.py ├── eval_func.py ├── log └── log.txt ├── model ├── loss_function │ ├── binary_cross_entropy.py │ ├── cross_entropy_loss.py │ ├── focal_loss.py │ ├── multilabel_cross_entropy.py │ └── span_loss.py ├── metrics │ └── metrics.py └── model.py ├── output └── result.json ├── predict.py ├── pretrained_model └── chinese_roberta_wwm_ext │ └── readme.txt ├── train.py └── utils ├── arguments_parse.py └── logger.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/README.md -------------------------------------------------------------------------------- /checkpoints/readme.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/data/test.json -------------------------------------------------------------------------------- /data/train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/data/train.json -------------------------------------------------------------------------------- /data_preprocessing/data_prepro.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/data_preprocessing/data_prepro.py -------------------------------------------------------------------------------- /data_preprocessing/tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/data_preprocessing/tools.py -------------------------------------------------------------------------------- /eval_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/eval_func.py -------------------------------------------------------------------------------- /log/log.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/loss_function/binary_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/model/loss_function/binary_cross_entropy.py -------------------------------------------------------------------------------- /model/loss_function/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/model/loss_function/cross_entropy_loss.py -------------------------------------------------------------------------------- /model/loss_function/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/model/loss_function/focal_loss.py -------------------------------------------------------------------------------- /model/loss_function/multilabel_cross_entropy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/model/loss_function/multilabel_cross_entropy.py -------------------------------------------------------------------------------- /model/loss_function/span_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/model/loss_function/span_loss.py -------------------------------------------------------------------------------- /model/metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/model/metrics/metrics.py -------------------------------------------------------------------------------- /model/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/model/model.py -------------------------------------------------------------------------------- /output/result.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/output/result.json -------------------------------------------------------------------------------- /predict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/predict.py -------------------------------------------------------------------------------- /pretrained_model/chinese_roberta_wwm_ext/readme.txt: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/train.py -------------------------------------------------------------------------------- /utils/arguments_parse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/utils/arguments_parse.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/suolyer/PyTorch_BERT_MultiHead_NER/HEAD/utils/logger.py --------------------------------------------------------------------------------