Skip to main content

Spello: Fast and Smart Spell Correction

Project description

A Fast and Accurate SpellCorrection using Sound and Edit-distance based Correction available in English and Hindi language.

GitHub stars Downloads Pypi package GitHub issues GitHub license Contributors

What is itInstallationGetting Started

What is it

Spello is a spellcorrection model built with combination of two models, Phoneme and Symspell Phoneme Model uses Soundex algo in background and suggests correct spellings using phonetic concepts to identify similar sounding words. On the other hand, Symspell Model uses concept of edit-distance in order to suggest correct spellings. Spello get's you best of both, taking into consideration context of the word as well.
Currently, this module is available for English(en) and Hindi(hi).

💾 Installation

▴ Back to top

Install the spello via `pip`
$ pip install spello

You can either train a new model from scratch or use pre-trained model. Alternatively you can also train model for your domain and use that on priority while use pre-trained model as a fallback

⚡ ️Getting Started

▴ Back to top

1. Model Initialisation

Initialise the model for one of the suppored languages.

>>> from spello.model import SpellCorrectionModel  
>>> sp = SpellCorrectionModel(language='en')  

2. Model Training - New Model

You can choose to train model by providing data in one of the following format

  • List of text or sentences.
  • Dict having word and their corresponding count.

Training providing list of sentences

>>> sp.train(['I want to play cricket', 'this is a text corpus'])

Training providing words counter

>>> sp.train({'i': 2, 'want': 1, 'play': 1, 'cricket': 10, 'mumbai': 5})

List of text is a recommended type for training data as here model also tries to learn context in which words are appearing, which further help to find best possible suggestion in case more than one suggestions are suggested by symspell or phoneme model

3. Model Prediction

>>> sp.spell_correct('i wnt to plai kricket')  
{'original_text': 'i wnt to plai kricket',
 'spell_corrected_text': 'i want to play cricket',
 'correction_dict': {'wnt': 'want', 'plai': 'play', 'kricket': 'cricket'}
}

4. Save Model

Call the save method to save the trained model at given model dir

>>> sp.save(model_save_dir='/home/ubuntu/')
'/home/ubuntu/model.pkl' # saved model path

5. Load Model

Load the trained model from saved path, First initialise the model and call the load method

>>> from spello.model import SpellCorrectionModel
>>> sp = SpellCorrectionModel(language='en')
>>> sp.load('/home/ubuntu/model.pkl')

6. Customize Configuration of Model (Optional)

Here, you are also provided to customize various configuration of the model like

  1. Setting minumum and maximum length eligible for spellcorrection
>>> sp.config.min_length_for_spellcorrection = 2 # default is 3
>>> sp.config.max_length_for_spellcorrection = 20 # default is 15
  1. Setting Max edit distance allowed for each char level for symspell and phoneme model
>>> sp.config.symspell_allowed_distance_map = {2:0, 3: 1, 4: 2, 5: 3, 6: 3, 7: 4, 8: 4, 9:5, 10:5, 11:5, 12:5, 13: 6, 14: 6, 15: 6, 16: 6, 17: 6, 18: 6, 19: 6, 20: 6}
# above dict signifies max edit distance possible for word of length 6 is 3, for length 7 is 4 and so on..

To reset to default config

>>> sp.set_default_config()

there are many more configurations which you can set, check this file for more details

Get Started with Pre-trained Models

We have trained a basic model on 30K news + 30k wikipedia sentences
Follow below steps to get started with these model

  1. Download model from below link

  2. Unzip the downloaded file

  3. Init and Load the model by specifying path of unzipped file

>>> from spello.model import SpellCorrectionModel
>>> sp = SpellCorrectionModel(language='en')
>>> sp.load('/path/to/file/en.pkl')
  1. Run the spell correction
>>> sp.spell_correct('i wnt to ply futbal')
{'original_text': 'i wnt to ply futbal',
 'spell_corrected_text': 'i want to play football',
 'correction_dict': {'wnt': 'want', 'ply': 'play', 'futbal': 'football'}
}

To train model for other languages, you can download data from here and follow training process.

Credits

This software uses the following open source packages:

Contribution guidelines

This project follows the all-contributors specification. Contributions of any kind welcome!

Please read the contribution guidelines first.

Future Scope / Limitations

One of the limitations of the current model is, it does not suggest corrections for any grammatical mistakes or for words in the vocabulary of the model. For example, in a sentence “I want to by Apple”, it will not suggest any correction for “by” as it is a valid English word but the correct replacement should be "buy".

In a future release, we will be adding features to suggest corrections for improper use word in a sentence.

Citing

▴ Back to top

If you use spello in a scientific publication, we would appreciate references to the following BibTex entry:

@misc{haptik2020spello,
  title={spello},
  author={Srivastava, Aman},
  howpublished={\url{https://github.com/hellohaptik/spello}},
  year={2020}
}

Project details


Download files

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

Files for spello, version 1.2.0
Filename, size File type Python version Upload date Hashes
Filename, size spello-1.2.0-py3-none-any.whl (35.0 kB) File type Wheel Python version py3 Upload date Hashes View hashes
Filename, size spello-1.2.0.tar.gz (24.0 kB) File type Source Python version None Upload date Hashes View hashes

Supported by

Elastic Elastic Search Pingdom Pingdom Monitoring Google Google BigQuery Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page