Skip to main content

Hidden alignment conditional random field, a discriminative string edit distance

Project description

https://travis-ci.org/dedupeio/pyhacrf.svg?branch=master https://ci.appveyor.com/api/projects/status/kibqrd7wnsk2ilpf/branch/master?svg=true

Hidden alignment conditional random field for classifying string pairs - a learnable edit distance.

Part of the Dedupe.io cloud service and open source toolset for de-duplicating and finding fuzzy matches in your data: https://dedupe.io

This package aims to implement the HACRF machine learning model with a sklearn-like interface. It includes ways to fit a model to training examples and score new example.

The model takes string pairs as input and classify them into any number of classes. In McCallum’s original paper the model was applied to the database deduplication problem. Each database entry was paired with every other entry and the model then classified whether the pair was a ‘match’ or a ‘mismatch’ based on training examples of matches and mismatches.

I also tried to use it as learnable string edit distance for normalizing noisy text. See A Conditional Random Field for Discriminatively-trained Finite-state String Edit Distance by McCallum, Bellare, and Pereira, and the report Conditional Random Fields for Noisy text normalisation by Dirko Coetsee.

Example

from pyhacrf import StringPairFeatureExtractor, Hacrf

training_X = [('helloooo', 'hello'), # Matching examples
              ('h0me', 'home'),
              ('krazii', 'crazy'),
              ('non matching string example', 'no really'), # Non-matching examples
              ('and another one', 'yep')]
training_y = ['match',
              'match',
              'match',
              'non-match',
              'non-match']

# Extract features
feature_extractor = StringPairFeatureExtractor(match=True, numeric=True)
training_X_extracted = feature_extractor.fit_transform(training_X)

# Train model
model = Hacrf(l2_regularization=1.0)
model.fit(training_X_extracted, training_y)

# Evaluate
from sklearn.metrics import confusion_matrix
predictions = model.predict(training_X_extracted)

print(confusion_matrix(training_y, predictions))
> [[0 3]
>  [2 0]]

print(model.predict_proba(training_X_extracted))
> [[ 0.94914812  0.05085188]
>  [ 0.92397711  0.07602289]
>  [ 0.86756034  0.13243966]
>  [ 0.05438812  0.94561188]
>  [ 0.02641275  0.97358725]]

Dependencies

This package depends on numpy. The LBFGS optimizer in pylbfgs is used, but alternative optimizers can be passed.

Install

Install by running:

python setup.py install

or from pypi:

pip install pyhacrf

Developing

Clone from repository, then

pip install -r requirements.txt
cython pyhacrf/*.pyx
python setup.py install

To deploy to pypi, make sure you have compiled the *.pyx files to *.c

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

pyhacrf-datamade-0.2.7.tar.gz (352.7 kB view hashes)

Uploaded Source

Built Distributions

pyhacrf_datamade-0.2.7-pp310-pypy310_pp73-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (255.1 kB view hashes)

Uploaded PyPy manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-pp310-pypy310_pp73-manylinux_2_17_i686.manylinux2014_i686.whl (261.6 kB view hashes)

Uploaded PyPy manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-pp310-pypy310_pp73-macosx_10_9_x86_64.whl (194.3 kB view hashes)

Uploaded PyPy macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-pp39-pypy39_pp73-win_amd64.whl (183.9 kB view hashes)

Uploaded PyPy Windows x86-64

pyhacrf_datamade-0.2.7-pp39-pypy39_pp73-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (249.8 kB view hashes)

Uploaded PyPy manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-pp39-pypy39_pp73-manylinux_2_17_i686.manylinux2014_i686.whl (255.6 kB view hashes)

Uploaded PyPy manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-pp39-pypy39_pp73-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (242.9 kB view hashes)

Uploaded PyPy manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-pp39-pypy39_pp73-macosx_10_9_x86_64.whl (191.2 kB view hashes)

Uploaded PyPy macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-pp38-pypy38_pp73-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (252.0 kB view hashes)

Uploaded PyPy manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-pp37-pypy37_pp73-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (254.0 kB view hashes)

Uploaded PyPy manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp312-cp312-win_amd64.whl (192.4 kB view hashes)

Uploaded CPython 3.12 Windows x86-64

pyhacrf_datamade-0.2.7-cp312-cp312-win32.whl (162.3 kB view hashes)

Uploaded CPython 3.12 Windows x86

pyhacrf_datamade-0.2.7-cp312-cp312-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2 MB view hashes)

Uploaded CPython 3.12 manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp312-cp312-manylinux_2_17_i686.manylinux2014_i686.whl (1.2 MB view hashes)

Uploaded CPython 3.12 manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-cp312-cp312-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.2 MB view hashes)

Uploaded CPython 3.12 manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-cp312-cp312-macosx_11_0_arm64.whl (199.6 kB view hashes)

Uploaded CPython 3.12 macOS 11.0+ ARM64

pyhacrf_datamade-0.2.7-cp312-cp312-macosx_10_9_x86_64.whl (221.9 kB view hashes)

Uploaded CPython 3.12 macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-cp312-cp312-macosx_10_9_universal2.whl (410.0 kB view hashes)

Uploaded CPython 3.12 macOS 10.9+ universal2 (ARM64, x86-64)

pyhacrf_datamade-0.2.7-cp311-cp311-win_amd64.whl (189.9 kB view hashes)

Uploaded CPython 3.11 Windows x86-64

pyhacrf_datamade-0.2.7-cp311-cp311-win32.whl (160.4 kB view hashes)

Uploaded CPython 3.11 Windows x86

pyhacrf_datamade-0.2.7-cp311-cp311-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.3 MB view hashes)

Uploaded CPython 3.11 manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp311-cp311-manylinux_2_17_i686.manylinux2014_i686.whl (1.2 MB view hashes)

Uploaded CPython 3.11 manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-cp311-cp311-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.2 MB view hashes)

Uploaded CPython 3.11 manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-cp311-cp311-macosx_11_0_arm64.whl (196.1 kB view hashes)

Uploaded CPython 3.11 macOS 11.0+ ARM64

pyhacrf_datamade-0.2.7-cp311-cp311-macosx_10_9_x86_64.whl (221.1 kB view hashes)

Uploaded CPython 3.11 macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-cp311-cp311-macosx_10_9_universal2.whl (406.3 kB view hashes)

Uploaded CPython 3.11 macOS 10.9+ universal2 (ARM64, x86-64)

pyhacrf_datamade-0.2.7-cp310-cp310-win_amd64.whl (194.0 kB view hashes)

Uploaded CPython 3.10 Windows x86-64

pyhacrf_datamade-0.2.7-cp310-cp310-win32.whl (166.1 kB view hashes)

Uploaded CPython 3.10 Windows x86

pyhacrf_datamade-0.2.7-cp310-cp310-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp310-cp310-manylinux_2_17_i686.manylinux2014_i686.whl (1.1 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-cp310-cp310-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.2 MB view hashes)

Uploaded CPython 3.10 manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-cp310-cp310-macosx_11_0_arm64.whl (196.3 kB view hashes)

Uploaded CPython 3.10 macOS 11.0+ ARM64

pyhacrf_datamade-0.2.7-cp310-cp310-macosx_10_9_x86_64.whl (222.3 kB view hashes)

Uploaded CPython 3.10 macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-cp310-cp310-macosx_10_9_universal2.whl (407.7 kB view hashes)

Uploaded CPython 3.10 macOS 10.9+ universal2 (ARM64, x86-64)

pyhacrf_datamade-0.2.7-cp39-cp39-win_amd64.whl (192.9 kB view hashes)

Uploaded CPython 3.9 Windows x86-64

pyhacrf_datamade-0.2.7-cp39-cp39-win32.whl (165.3 kB view hashes)

Uploaded CPython 3.9 Windows x86

pyhacrf_datamade-0.2.7-cp39-cp39-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2 MB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp39-cp39-manylinux_2_17_i686.manylinux2014_i686.whl (1.1 MB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-cp39-cp39-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.2 MB view hashes)

Uploaded CPython 3.9 manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-cp39-cp39-macosx_11_0_arm64.whl (197.0 kB view hashes)

Uploaded CPython 3.9 macOS 11.0+ ARM64

pyhacrf_datamade-0.2.7-cp39-cp39-macosx_10_9_x86_64.whl (220.7 kB view hashes)

Uploaded CPython 3.9 macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-cp39-cp39-macosx_10_9_universal2.whl (406.7 kB view hashes)

Uploaded CPython 3.9 macOS 10.9+ universal2 (ARM64, x86-64)

pyhacrf_datamade-0.2.7-cp38-cp38-win_amd64.whl (193.5 kB view hashes)

Uploaded CPython 3.8 Windows x86-64

pyhacrf_datamade-0.2.7-cp38-cp38-win32.whl (165.5 kB view hashes)

Uploaded CPython 3.8 Windows x86

pyhacrf_datamade-0.2.7-cp38-cp38-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.2 MB view hashes)

Uploaded CPython 3.8 manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp38-cp38-manylinux_2_17_i686.manylinux2014_i686.whl (1.1 MB view hashes)

Uploaded CPython 3.8 manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-cp38-cp38-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.2 MB view hashes)

Uploaded CPython 3.8 manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-cp38-cp38-macosx_11_0_arm64.whl (196.5 kB view hashes)

Uploaded CPython 3.8 macOS 11.0+ ARM64

pyhacrf_datamade-0.2.7-cp38-cp38-macosx_10_9_x86_64.whl (219.4 kB view hashes)

Uploaded CPython 3.8 macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-cp38-cp38-macosx_10_9_universal2.whl (404.6 kB view hashes)

Uploaded CPython 3.8 macOS 10.9+ universal2 (ARM64, x86-64)

pyhacrf_datamade-0.2.7-cp37-cp37m-win_amd64.whl (192.4 kB view hashes)

Uploaded CPython 3.7m Windows x86-64

pyhacrf_datamade-0.2.7-cp37-cp37m-win32.whl (163.4 kB view hashes)

Uploaded CPython 3.7m Windows x86

pyhacrf_datamade-0.2.7-cp37-cp37m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.1 MB view hashes)

Uploaded CPython 3.7m manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp37-cp37m-manylinux_2_17_i686.manylinux2014_i686.whl (1.0 MB view hashes)

Uploaded CPython 3.7m manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-cp37-cp37m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.1 MB view hashes)

Uploaded CPython 3.7m manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-cp37-cp37m-macosx_10_9_x86_64.whl (220.0 kB view hashes)

Uploaded CPython 3.7m macOS 10.9+ x86-64

pyhacrf_datamade-0.2.7-cp36-cp36m-manylinux_2_17_x86_64.manylinux2014_x86_64.whl (1.1 MB view hashes)

Uploaded CPython 3.6m manylinux: glibc 2.17+ x86-64

pyhacrf_datamade-0.2.7-cp36-cp36m-manylinux_2_17_i686.manylinux2014_i686.whl (1.0 MB view hashes)

Uploaded CPython 3.6m manylinux: glibc 2.17+ i686

pyhacrf_datamade-0.2.7-cp36-cp36m-manylinux_2_17_aarch64.manylinux2014_aarch64.whl (1.1 MB view hashes)

Uploaded CPython 3.6m manylinux: glibc 2.17+ ARM64

pyhacrf_datamade-0.2.7-cp36-cp36m-macosx_10_9_x86_64.whl (214.3 kB view hashes)

Uploaded CPython 3.6m macOS 10.9+ x86-64

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