HUSTAI/uie_pytorch
A PyTorch implementation of UIE, a universal model for information extraction tasks including NER, relation extraction, and event extraction.

This repository ports PaddleNLP’s UIE model to PyTorch. UIE provides a unified framework for multiple NLP extraction tasks such as named entity recognition, relation extraction, event extraction, and sentiment classification. The implementation includes fine-tuning scripts, evaluation tools, ONNX model export for deployment, and inference predictors. It supports various information extraction scenarios out of the box and offers ErnieM tokenizer integration for improved performance.
Frequently asked
- What is HUSTAI/uie_pytorch?
- A PyTorch implementation of UIE, a universal model for information extraction tasks including NER, relation extraction, and event extraction.
- Is uie_pytorch open source?
- Yes — HUSTAI/uie_pytorch is open source, released under the Apache-2.0 license.
- What language is uie_pytorch written in?
- HUSTAI/uie_pytorch is primarily written in Python.
- How popular is uie_pytorch?
- HUSTAI/uie_pytorch has 702 stars on GitHub.
- Where can I find uie_pytorch?
- HUSTAI/uie_pytorch is on GitHub at https://github.com/HUSTAI/uie_pytorch.