Fast English word segmentation in Rust
Go to file
Michael Partheil 714b21aad5 Pre-compute unigram and bigram scores 2023-10-17 14:42:50 +02:00
.cargo py: initial version of Python bindings 2021-03-24 11:57:29 +01:00
.github Use string for Python 3.10 2022-09-13 12:19:43 +02:00
data Remove old data files 2021-06-03 16:09:24 +02:00
instant-segment Pre-compute unigram and bigram scores 2023-10-17 14:42:50 +02:00
instant-segment-py Upgrade pyo3 to 0.20 2023-10-12 16:07:18 +02:00
.gitignore Add script to download word list input data 2021-06-03 16:09:24 +02:00
Cargo.toml py: initial version of Python bindings 2021-03-24 11:57:29 +01:00
LICENSE Add license files (fixes #15) 2021-04-29 15:34:02 +02:00
Makefile py: initial version of Python bindings 2021-03-24 11:57:29 +01:00
README.md Update links to point to new GitHub org 2021-08-31 14:10:13 +02:00
cover.svg Add files via upload 2021-04-20 11:06:47 -07:00
deny.toml Allow MPL licenses 2020-11-25 17:34:03 +01:00

README.md

Cover logo

Instant Segment: fast English word segmentation in Rust

Documentation Crates.io PyPI Build status License: Apache 2.0

Instant Segment is a fast Apache-2.0 library for English word segmentation. It is based on the Python wordsegment project written by Grant Jenks, which is in turn based on code from Peter Norvig's chapter Natural Language Corpus Data from the book Beautiful Data (Segaran and Hammerbacher, 2009).

For the microbenchmark included in this repository, Instant Segment is ~100x faster than the Python implementation. The API was carefully constructed so that multiple segmentations can share the underlying state to allow parallel usage.

How it works

Instant Segment works by segmenting a string into words by selecting the splits with the highest probability given a corpus of words and their occurrences.

For instance, provided that choose and spain occur more frequently than chooses and pain, and that the pair choose spain occurs more frequently than chooses pain, Instant Segment can help identify the domain choosespain.com as ChooseSpain.com which more likely matches user intent.

Read about how we built and improved Instant Segment for use in production at Instant Domain Search to help our users find relevant domains they can register.

Using the library

Python (>= 3.9)

pip install instant-segment

Rust

[dependencies]
instant-segment = "0.8.1"

Examples

The following examples expect unigrams and bigrams to exist. See the examples (Rust, Python) to see how to construct these objects.

import instant_segment

segmenter = instant_segment.Segmenter(unigrams, bigrams)
search = instant_segment.Search()
segmenter.segment("instantdomainsearch", search)
print([word for word in search])

--> ['instant', 'domain', 'search']
use instant_segment::{Search, Segmenter};
use std::collections::HashMap;

let segmenter = Segmenter::from_maps(unigrams, bigrams);
let mut search = Search::default();
let words = segmenter
    .segment("instantdomainsearch", &mut search)
    .unwrap();
println!("{:?}", words.collect::<Vec<&str>>())

--> ["instant", "domain", "search"]

Check out the tests for more thorough examples: Rust, Python

Testing

To run the tests run the following:

cargo t -p instant-segment --all-features

You can also test the Python bindings with:

make test-python