Skip to main content

Vose-Alias-Method

Python implementation of Vose's alias method, an efficient algorithm for sampling from a discrete probability distribution (a good explanation of which can be found at http://www.keithschwarz.com/darts-dice-coins/).

For example, this code can be used for creating and efficiently sampling from a probability distribution representing rolling a weighted die (i.e where side j has probability P(j) of being rolled). Alternatively, it could be used for creating a simple unigram language model (see example below)

Any suggestions/contributions very welcome.

Installation

$ pip install Vose-Alias-Method

Or via conda: $ conda install -c conda-forge vose-alias-method

Depends on:

Example Usage

In a python shell:

>>> from vose_sampler import VoseAlias
>>> # Create the required probability distribution (here we use the example of a weighted coin with probability H:=Heads=0.2 and T:=Tail=0.8)
>>> dist = {"H":0.2, "T":0.8}
>>> # Create probability and alias tables from the probability distribution, for sampling via Vose's alias method
>>> VA = VoseAlias(dist)
>>> # Generate n random outcomes (here n=10)
>>> VA.sample_n(size=10)
['T', 'T', 'H', 'T', 'T', 'T', 'T', 'H', 'T', 'T']

Unigram language model example

To create a unigram language model for Alice in Wonderland and sample 10 words from this, run the main script from the command line with options:

$ vose-sampler -p data/Alice.txt -n 10  # or: python vose_sampler/vose_sampler.py -p data/Alice.txt -n 10

Generating 10 random samples:

the
more
she
Rabbit,
say
suddenly
at
soon
thing
solemn

[Note, this is intended to illustrate how Vose's alias method could be used. Thus I have not included any preprocessing steps that would make the language model more realistic; for example, we could add handling of upper vs. lower case words (so that e.g. "The" and "the" are not considered distinct), as well as handling of punctuation (e.g. so "the" and "the." are considered the same).]

Tests

Run via: $ python setup.py test (or $ python tests/tests.py)

Build

A GitHub Action exists for this at: .github/workflows/publish.yaml

Alternatively build and publish manually with:

  • $ python setup.py sdist bdist_wheel
  • $ twine upload dist/* -r testpypi --skip-existing assuming twine is installed and ~/.pypirc exists with something like:
[distutils]
index-servers=
    testpypi
    pypi
    
[testpypi]
repository = https://test.pypi.org/legacy/
username = asmith26
password = some_password

[pypi]
repository = https://upload.pypi.org/legacy/
username = asmith26
password = some_harder_password
  • Assuming everything looks good $ twine upload dist/*
  • Create new git release $ git tag <tagname> && git push origin <tag_name>, and create a new release with the same <tagname>.

Download files

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

Source Distribution

vose_alias_method-1.2.2.tar.gz (10.6 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

vose_alias_method-1.2.2-py3-none-any.whl (13.8 kB view details)

Uploaded Python 3

File details

Details for the file vose_alias_method-1.2.2.tar.gz.

File metadata

  • Download URL: vose_alias_method-1.2.2.tar.gz
  • Upload date:
  • Size: 10.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for vose_alias_method-1.2.2.tar.gz
Algorithm Hash digest
SHA256 1333655259733da6241c89e256d75ccd055c326b9e262ec3151c792048b3316b
MD5 cd5d673cca3f7e8c99b7a504c284bd66
BLAKE2b-256 6968d30092fcec380844f32a9da56a0d5a0bd68a9362055a1774e1765025e8cd

See more details on using hashes here.

Provenance

The following attestation bundles were made for vose_alias_method-1.2.2.tar.gz:

Publisher: publish.yaml on asmith26/Vose-Alias-Method

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file vose_alias_method-1.2.2-py3-none-any.whl.

File metadata

File hashes

Hashes for vose_alias_method-1.2.2-py3-none-any.whl
Algorithm Hash digest
SHA256 edcf6ad8bc469c877bc5e031b544c6698d08bd1da693227ab92fb7110c43a59a
MD5 7b3da464dca66636065efd27e0af978a
BLAKE2b-256 dcae5b91e98dd04bbd5bdc0b470095874d7a62d6822bf753f32da6f18289b86c

See more details on using hashes here.

Provenance

The following attestation bundles were made for vose_alias_method-1.2.2-py3-none-any.whl:

Publisher: publish.yaml on asmith26/Vose-Alias-Method

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page