kmkurn / pytorch-crf

(Linear-chain) Conditional random field in PyTorch.
https://pytorch-crf.readthedocs.io
MIT License
943 stars 152 forks source link

How can we get the probabilities out? #36

Closed cinjon closed 2 years ago

cinjon commented 5 years ago

I see how to get the output of the CRF, as well as the decoding - is it possible to also get the computed probabilities for each of the predictions?

kmkurn commented 5 years ago

Unfortunately no. You can tweak the code yourself if you want. See #21 for a reference implementation.