← all repositories

mandarjoshi90/coref

An implementation of BERT and SpanBERT transformer models for coreference resolution, achieving up to 79.6 F1 on OntoNotes.

454 stars Python Language Models
coref
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This repository provides code and pretrained models for coreference resolution using BERT and SpanBERT. It extends the e2e-coref architecture with transformer-based encoders, enabling span-based mention detection and clustering for identifying all expressions referring to the same entity in text. The models are fine-tuned on the OntoNotes dataset and can be used via pretrained downloads or trained from scratch with GPU resources.

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