Skip to content

Add sequence probability support for PyTorch CRF model #421

@vrdn-23

Description

@vrdn-23

For active learning, there a couple of strategies that we currently utilize using the CRF suite model's marginal probabilities. Studies have shown that returning sequence-level probabilities instead of token-level marginal probabilities works much better and this is something that can be implemented in a future release. So in order to modify this function, in addition to the best transition score (t,j) and corresponding backward link for the transition, we’ll have to store top-n transition scores and the corresponding n backward links, and then trace all n-paths, resulting in the n-best sequences and corresponding probabilities.

Metadata

Metadata

Assignees

No one assigned

    Labels

    enhancementNew feature or request

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions