├── .gitignore ├── README.md ├── args.py ├── assets └── overview.jpg ├── config_reader.py ├── configs ├── ace2004.conf ├── ace2005.conf ├── conll03.conf ├── eval.conf ├── genia.conf ├── ontonotes.conf └── zhmsra.conf ├── diffusionner.py ├── diffusionner ├── __init__.py ├── diffusionner_trainer.py ├── entities.py ├── evaluator.py ├── input_reader.py ├── lap.py ├── loss.py ├── matcher.py ├── modeling_albert.py ├── modeling_bert.py ├── modeling_roberta.py ├── modeling_xlm_roberta.py ├── models.py ├── sampling.py ├── templates │ └── entity_examples.html ├── trainer.py └── util.py ├── requirements.txt └── tools ├── context_convert.py └── ner2json.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/args.py -------------------------------------------------------------------------------- /assets/overview.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/assets/overview.jpg -------------------------------------------------------------------------------- /config_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/config_reader.py -------------------------------------------------------------------------------- /configs/ace2004.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/configs/ace2004.conf -------------------------------------------------------------------------------- /configs/ace2005.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/configs/ace2005.conf -------------------------------------------------------------------------------- /configs/conll03.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/configs/conll03.conf -------------------------------------------------------------------------------- /configs/eval.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/configs/eval.conf -------------------------------------------------------------------------------- /configs/genia.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/configs/genia.conf -------------------------------------------------------------------------------- /configs/ontonotes.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/configs/ontonotes.conf -------------------------------------------------------------------------------- /configs/zhmsra.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/configs/zhmsra.conf -------------------------------------------------------------------------------- /diffusionner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner.py -------------------------------------------------------------------------------- /diffusionner/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diffusionner/diffusionner_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/diffusionner_trainer.py -------------------------------------------------------------------------------- /diffusionner/entities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/entities.py -------------------------------------------------------------------------------- /diffusionner/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/evaluator.py -------------------------------------------------------------------------------- /diffusionner/input_reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/input_reader.py -------------------------------------------------------------------------------- /diffusionner/lap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/lap.py -------------------------------------------------------------------------------- /diffusionner/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/loss.py -------------------------------------------------------------------------------- /diffusionner/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/matcher.py -------------------------------------------------------------------------------- /diffusionner/modeling_albert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/modeling_albert.py -------------------------------------------------------------------------------- /diffusionner/modeling_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/modeling_bert.py -------------------------------------------------------------------------------- /diffusionner/modeling_roberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/modeling_roberta.py -------------------------------------------------------------------------------- /diffusionner/modeling_xlm_roberta.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/modeling_xlm_roberta.py -------------------------------------------------------------------------------- /diffusionner/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/models.py -------------------------------------------------------------------------------- /diffusionner/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/sampling.py -------------------------------------------------------------------------------- /diffusionner/templates/entity_examples.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/templates/entity_examples.html -------------------------------------------------------------------------------- /diffusionner/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/trainer.py -------------------------------------------------------------------------------- /diffusionner/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/diffusionner/util.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/context_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/tools/context_convert.py -------------------------------------------------------------------------------- /tools/ner2json.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/tricktreat/DiffusionNER/HEAD/tools/ner2json.py --------------------------------------------------------------------------------