chaine 0.1.0

A lightweight Linear-Chain Conditional Random Field

A lightweight Linear-Chain Conditional Random Field

This is a modern, fast and no-dependency Python library implementing a linear-chain conditional random field for natural language processing tasks like named entity recognition or part-of-speech tagging.

Installation

You can install the latest stable version from PyPI:

$ pip install chaine

Example

>>> import chaine
>>> import datasets
>>> data = datasets.load_dataset("germeval_14")
>>> tokens = data["train"]["tokens"]
>>> labels = data["train"]["ner_tags"]
>>> crf = chaine.train(tokens, labels, max_iterations=100)
>>> sequence = chaine.featurize(["todo", "todo", "todo"])
>>> crf.predict(sequence)
["O", "O", "B-PER"]

Disclaimer

This library makes use of and is partially based on:

• CRFsuite
• libLBFGS
• python-crfsuite
• sklearn-crfsuite