Skip to main content

"Partial/Fuzzy Conditional random field in PyTorch."

Project description

pytorch-partial-crf

Partial/Fuzzy conditional random field in PyTorch.

How to use

Install

pip install pytorch-partial-crf

Use partial CRF

import torch
from pytorch_partial_crf import PartialCRF

# Create 
num_tags = 6
model = PartialCRF(num_tags)

batch_size, sequence_length = 3, 5
emissions = torch.randn(batch_size, sequence_length, num_tags)

# Set unknown tag to -1
tags = torch.LongTensor([
    [1, 2, 3, 3, 5],
    [-1, 3, 3, 2, -1],
    [-1, 0, -1, -1, 4],
])

# Computing log likelihood
model(emissions, tags)

License

MIT

References

The implementation is based on AllenNLP CRF module and pytorch-crf.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pytorch-partial-crf-0.0.6.tar.gz (5.2 kB view hashes)

Uploaded Source

Built Distribution

pytorch_partial_crf-0.0.6-py3-none-any.whl (10.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page