← all repositories

HHousen/TransformerSum

A library for training and evaluating transformer-based models on automatic text summarization tasks.

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

TransformerSum provides implementations of both extractive and abstractive summarization models using pre-trained transformer architectures. It integrates with Hugging Face Transformers for easy access to various model architectures and uses PyTorch Lightning to handle training loops. The library also includes utilities for converting abstractive summarization datasets into extractive format, enabling comparative research between the two approaches.

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