search

yixinL7 / SimCLS

Code for our paper "SimCLS: A Simple Framework for Contrastive Learning of Abstractive Summarization", ACL 2021
183 stars 25 forks source link

Where are candidate summaries created #12

Open crichardson332 opened 2 years ago

crichardson332 commented 2 years ago

Hello, I'm trying to train and evaluate this model on a new dataset (SAMSum). To do that I need to first generate and score candidate summaries using the generative model. Where is the code that was used to do this? The README for this repo seems like it starts with a dataset that already includes candidate summaries and their rogue scores. I just want to make sure I do that step correctly. Thanks!

yixinL7 commented 2 years ago

Hi! Please check out this file for a reference: https://github.com/yixinL7/BRIO/blob/main/gen_candidate.py