Conditional random field in PyTorch.
- Python 3.6
- PyTorch 0.3.0
You can install with pip
pip install pytorch-crf
Or, you can install from Github directly
pip install git+https://github.com/kmkurn/pytorch-crf#egg=pytorch_crf
In the examples below, we will assume that these lines have been executed
>>> import torch >>> from torchcrf import CRF >>> seq_length, batch_size, num_tags = 3, 2, 5 >>> emissions = torch.autograd.Variable(torch.randn(seq_length, batch_size, num_tags), requires_grad=True) >>> tags = torch.autograd.Variable(torch.LongTensor([[0, 1], [2, 4], [3, 1]])) # (seq_length, batch_size) >>> model = CRF(num_tags)
Computing log likelihood
>>> model(emissions, tags) Variable containing: -10.0635 [torch.FloatTensor of size 1]
Computing log likelihood with mask
>>> mask = torch.autograd.Variable(torch.ByteTensor([[1, 1], [1, 1], [1, 0]])) # (seq_length, batch_size) >>> model(emissions, tags, mask=mask) Variable containing: -8.4981 [torch.FloatTensor of size 1]
>>> model.decode(emissions) [[3, 1, 3], [0, 1, 0]]
Decoding with mask
>>> model.decode(emissions, mask=mask) [[3, 1, 3], [0, 1]]
tests/test_crf.py for more examples.
MIT. See LICENSE for details.
Contributions are welcome! Please follow these instructions to setup dependencies and running the tests and linter. Make a pull request once your contribution is ready.
Make sure you setup a virtual environment with Python 3.6 and PyTorch installed. Then, install all the dependencies in
requirements.txt file and install this package in development mode.
pip install -r requirements.txt pip install -e .
pytest in the project root directory.
flake8 in the project root directory. This will also run
mypy, thanks to