Skip to content

Fine-tuning the T5-base model using parts of the CNN/DailyMail dataset using PyTorch Lightning.

License

Notifications You must be signed in to change notification settings

MichalSuchocki/T5-text-summarization

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

T5-text-summarization

Fine-tuning the T5-base model using parts of the CNN/DailyMail dataset using PyTorch Lightning. Model research as part of a diploma thesis. When tested on a set of 150 samples, the following results were achieved:

Metrics Result
ROUGE-1 0.3874
ROUGE-2 0.1621
ROUGE-L 0.2708

About

Fine-tuning the T5-base model using parts of the CNN/DailyMail dataset using PyTorch Lightning.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published