allanj/pytorch_neural_crf

Is there a way to output not only crf 1 best, but 5 best predictions?

mxc19912008 opened this issue · 2 comments

mxc19912008 commented

Hi Allan,

Is there a way to output not only crf 1 best, but 5 best predictions? Any hint for this modification?

Thanks!

allanj commented

I'm currently on holiday until 10 Sept and will try to incorporate this function into this repo.
But in the meanwhile, you can check out this repo: https://github.com/jiesutd/NCRFpp/blob/master/model/crf.py#L268-L418

and see if this help

  • 👍3
  • ❤️1
mxc19912008 commented

Thanks! Have a good holiday!