princeton-nlp/PURE
PyTorch-based system that uses pre-trained language models to perform entity and relation extraction from text.

Velocity · 7d
+0.4
★ / day
Trend
→steady
star history
PURE implements a two-stage pipeline for entity and relation extraction: an entity model predicts all entities in a text passage simultaneously, followed by a relation model that considers every entity pair independently with typed entity markers to predict relation types. The system leverages pre-trained language models (PyTorch) and includes pre-trained models for ACE05 and SciERC datasets.