Skip to content

ksgr5566/News-Headline-Generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 

Repository files navigation

News-Headline-Generator

A news headline generator finetuned on T5-base, coded in pytorch lightning.

Trained checkpoint: best-checkpoint.ckpt

Finetuned on this HuggingFace Dataset, csv file.

Rouge-l: "r"-0.43 "p"-0.44 "f"-0.43

Semantic Similarity - 74.3 (cosine simiilarity)