BERT token level embedding with MxNet
Project description
Bert Embeddings
Install
pip install bert-embedding
If you want to run on GPU machine, please install mxnet-cu92
.
pip install mxnet-cu92
Usage
from bert_embedding import BertEmbedding
sentences = ["Hello World", "Token level embeddings from BERT model on mxnet and gluonnlp"]
bert = BertEmbedding()
result = bert.embedding(sentences)
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
bert_embedding-0.1.0.tar.gz
(9.3 kB
view details)
Built Distribution
File details
Details for the file bert_embedding-0.1.0.tar.gz
.
File metadata
- Download URL: bert_embedding-0.1.0.tar.gz
- Upload date:
- Size: 9.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/1.12.1 pkginfo/1.5.0.1 requests/2.18.4 setuptools/40.8.0 requests-toolbelt/0.9.1 tqdm/4.31.0 CPython/3.5.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | b9a0cb2f4fc47e2fe6260d05ce9b119a42215824c097f440a88068832e981494 |
|
MD5 | fe632ff3429d78675477a117d119d477 |
|
BLAKE2b-256 | b0cbf885c8f0b62e8763c568c2d36a3a67fa9b1d06a92e81a718aa322d2639f4 |
Provenance
File details
Details for the file bert_embedding-0.1.0-py3-none-any.whl
.
File metadata
- Download URL: bert_embedding-0.1.0-py3-none-any.whl
- Upload date:
- Size: 13.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/1.12.1 pkginfo/1.5.0.1 requests/2.18.4 setuptools/40.8.0 requests-toolbelt/0.9.1 tqdm/4.31.0 CPython/3.5.6
File hashes
Algorithm | Hash digest | |
---|---|---|
SHA256 | 4710fbd7f8f7a86669e1bf96c094fee4ed697064887b3dfd0bcf82f81f247f79 |
|
MD5 | eb711e5648965d59507bfefce9689225 |
|
BLAKE2b-256 | 607095d01fba440df1cfe1b6fa6b46510f0bbcec8be8c4f6ee192a429346271a |