search

BorealisAI / keyphrase-generation

PyTorch code of “Diverse Keyphrase Generation with Neural Unlikelihood Training” (COLING 2020)
Other
5 stars 2 forks source link

Results for baselines #1

Closed kgarg8 closed 3 years ago

kgarg8 commented 3 years ago

Hi,

Do you have have baseline model files e.g. catSeq, catSeqD, catSeqCorr, catSeqTG, catSeqTG2RF1? You must be using them for diversity evaluation.

HareeshBahuleyan commented 3 years ago

Hi @kgarg8, for training the baselines, we use the open-source code provided by Chan et al. (2019): https://github.com/kenchan0226/keyphrase-generation-rl

Once we have the output keyphrases on the testset, we run it through our evaluation scripts to obtain the quality and diversity metrics.

kgarg8 commented 3 years ago

Makes sense, thanks Hareesh for the reply.