← all repositories

nlpyang/BertSum

A research implementation that fine-tunes BERT for extractive document summarization.

1.5k stars Python Language ModelsML Frameworks
BertSum
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository implements the paper ‘Fine-tune BERT for Extractive Summarization’ by adding multiple sentence embeddings and interval segments to BERT, then fine-tuning for extractive summarization tasks. The code supports training on CNN/DailyMail datasets and evaluates using ROUGE metrics. It uses PyTorch as the deep learning framework and Stanford CoreNLP for preprocessing.

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