lorenzkuhn / semantic_uncertainty

MIT License
136 stars 19 forks source link

Generation of the most likely answers #5

Open seonwoo-min opened 1 year ago

seonwoo-min commented 1 year ago

Hi Lorenz,

Thanks for the great work and codes! While I look through the codes, I notice a possible discrepancy between the paper and the codes. The paper states that you used beam search with num_beams=5 and do_sample=True to obtain the most likely answers. On the other hand, the generate.py script is using the beam search with do_sample=False instead. Am I missing something here?

https://github.com/lorenzkuhn/semantic_uncertainty/blob/27adbf0dc1bf056c771c205d89c2a79cbd82dc3a/code/generate.py#L113-L130