Skip to main content

Lightweight package for reading/writing pre-trained word embedding files

Project description

# pyemblib

A module for reading, writing, and using trained word embeddings.

## Installation

Install with pip!

`bash pip install pyemblib `

## Usage

This package currently supports word embeddings trained by the following packages:

### Reading

Both text-format and binary embedding files are supported.

The example below shows reading each format of embedding: `python ## import text embeddings text_embs ='/tmp/text_embeddings.txt', mode=pyemblib.Mode.Text) ## import binary embeddings bin_embs ='/tmp/bin_embeddings.bin', mode=pyemblib.Mode.Binary) `

Embeddings are read as a pyemblib.Embeddings object, which inherits from Python’s dictionary class; keys are words, and values are the embedding arrays.

To get the word vector for “python”, just use dictionary access: `python vec = embs['python'] print(vec) # [ 0.001 -0.237 ... ] `

### Writing

The same text and binary modes can be used for writing out embedding files as for reading.

`python embs = { 'a' : np.array([0.3 0.1 -0.2]), 'b' : np.array([-0.9, -0.2, -0.2]) } ## write as text pyemblib.write(embs, '/tmp/text_embeddings.txt', mode=pyemblib.Mode.Text) ## write as binary pyemblib.write(embs, '/tmp/bin_embeddings.bin', mode=pyemblib.Mode.Binary) `

## Feedback Please report any issues you encounter to the [Github Issues page](!

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

pyemblib-0.1.2.tar.gz (8.8 kB view hashes)

Uploaded source

Built Distribution

pyemblib-0.1.2-py3-none-any.whl (12.6 kB view hashes)

Uploaded py3

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