← all repositories

HUSTAI/uie_pytorch

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

702 stars Python Language Models
uie_pytorch
Velocity · 7d
+0.5
★ / day
Trend
steady
star history

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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.