← all repositories

Jacen789/relation-extraction

BERT-powered Chinese relation extraction model classifying person-to-person relationships into 12 categories.

468 stars Python Language ModelsML Frameworks
relation-extraction
Velocity · 7d
+0.2
★ / day
Trend
steady
star history

This project implements a Chinese relation extraction system using a BERT-based neural network trained with PyTorch. The model classifies pairs of entities in Chinese text into relationship categories such as parent, spouse, teacher-student, siblings, and colleagues. It achieves 78% accuracy on a test set of 1000 samples after 20 training epochs.

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