├── .gitignore ├── README.md ├── data └── data.rar ├── data_loader ├── __init__.py ├── data_process.py ├── rel2id.json ├── token_type2id.json └── vec.txt ├── deploy ├── demo.py └── test.json ├── deploy_flask ├── manage.py ├── static │ ├── css │ │ └── nlp.css │ └── js │ │ ├── cytoscape.min.js │ │ └── jquery-2.1.4.min.js └── templates │ ├── index.html │ └── index_neo4j.html ├── doc └── img │ ├── deploy.png │ ├── entity.png │ ├── model.png │ ├── relation.png │ └── relation_extract.png ├── mains ├── __init__.py ├── trainer.py └── trainer_std.py ├── models ├── 43m-p0.89f0.91n0.94r0.43.pth └── albert33m-p0.77f0.77n2.98r2.17.pth ├── modules ├── __init__.py ├── joint_model.py └── joint_model_adv.py ├── pretrained ├── albert_chinese_tiny │ ├── README.md │ ├── config.json │ ├── pytorch_model.bin │ └── vocab.txt └── bert-base-chinese │ ├── config.json │ ├── tokenizer.json │ ├── tokenizer_config.json │ └── vocab.txt ├── requirements.txt ├── test └── py2neo_test.py └── utils ├── FocalLoss.py ├── __init__.py ├── arg_config.py ├── config.py └── neo4j_util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/README.md -------------------------------------------------------------------------------- /data/data.rar: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/data/data.rar -------------------------------------------------------------------------------- /data_loader/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/data_loader/__init__.py -------------------------------------------------------------------------------- /data_loader/data_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/data_loader/data_process.py -------------------------------------------------------------------------------- /data_loader/rel2id.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/data_loader/rel2id.json -------------------------------------------------------------------------------- /data_loader/token_type2id.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/data_loader/token_type2id.json -------------------------------------------------------------------------------- /data_loader/vec.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/data_loader/vec.txt -------------------------------------------------------------------------------- /deploy/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy/demo.py -------------------------------------------------------------------------------- /deploy/test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy/test.json -------------------------------------------------------------------------------- /deploy_flask/manage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy_flask/manage.py -------------------------------------------------------------------------------- /deploy_flask/static/css/nlp.css: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy_flask/static/css/nlp.css -------------------------------------------------------------------------------- /deploy_flask/static/js/cytoscape.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy_flask/static/js/cytoscape.min.js -------------------------------------------------------------------------------- /deploy_flask/static/js/jquery-2.1.4.min.js: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy_flask/static/js/jquery-2.1.4.min.js -------------------------------------------------------------------------------- /deploy_flask/templates/index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy_flask/templates/index.html -------------------------------------------------------------------------------- /deploy_flask/templates/index_neo4j.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/deploy_flask/templates/index_neo4j.html -------------------------------------------------------------------------------- /doc/img/deploy.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/doc/img/deploy.png -------------------------------------------------------------------------------- /doc/img/entity.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/doc/img/entity.png -------------------------------------------------------------------------------- /doc/img/model.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/doc/img/model.png -------------------------------------------------------------------------------- /doc/img/relation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/doc/img/relation.png -------------------------------------------------------------------------------- /doc/img/relation_extract.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/doc/img/relation_extract.png -------------------------------------------------------------------------------- /mains/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/mains/__init__.py -------------------------------------------------------------------------------- /mains/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/mains/trainer.py -------------------------------------------------------------------------------- /mains/trainer_std.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/mains/trainer_std.py -------------------------------------------------------------------------------- /models/43m-p0.89f0.91n0.94r0.43.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/models/43m-p0.89f0.91n0.94r0.43.pth -------------------------------------------------------------------------------- /models/albert33m-p0.77f0.77n2.98r2.17.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/models/albert33m-p0.77f0.77n2.98r2.17.pth -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/modules/__init__.py -------------------------------------------------------------------------------- /modules/joint_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/modules/joint_model.py -------------------------------------------------------------------------------- /modules/joint_model_adv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/modules/joint_model_adv.py -------------------------------------------------------------------------------- /pretrained/albert_chinese_tiny/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/pretrained/albert_chinese_tiny/README.md -------------------------------------------------------------------------------- /pretrained/albert_chinese_tiny/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/pretrained/albert_chinese_tiny/config.json -------------------------------------------------------------------------------- /pretrained/albert_chinese_tiny/pytorch_model.bin: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/pretrained/albert_chinese_tiny/pytorch_model.bin -------------------------------------------------------------------------------- /pretrained/albert_chinese_tiny/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/pretrained/albert_chinese_tiny/vocab.txt -------------------------------------------------------------------------------- /pretrained/bert-base-chinese/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/pretrained/bert-base-chinese/config.json -------------------------------------------------------------------------------- /pretrained/bert-base-chinese/tokenizer.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/pretrained/bert-base-chinese/tokenizer.json -------------------------------------------------------------------------------- /pretrained/bert-base-chinese/tokenizer_config.json: -------------------------------------------------------------------------------- 1 | { 2 | "do_lower_case": false 3 | } 4 | -------------------------------------------------------------------------------- /pretrained/bert-base-chinese/vocab.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/pretrained/bert-base-chinese/vocab.txt -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/requirements.txt -------------------------------------------------------------------------------- /test/py2neo_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/test/py2neo_test.py -------------------------------------------------------------------------------- /utils/FocalLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/utils/FocalLoss.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/arg_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/utils/arg_config.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/neo4j_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Xie-Minghui/MultiHeadJointEntityRelationExtraction_simple/HEAD/utils/neo4j_util.py --------------------------------------------------------------------------------