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.