Metadata-Version: 2.1
Name: bert-embedding
Version: 0.1.0
Summary: BERT token level embedding with MxNet
Home-page: https://github.com/imgarylai/bert_embedding
Author: Gary Lai
Author-email: gary@gary-lai.com
License: ALv2
Download-URL: https://github.com/imgarylai/bert_embedding/tree/master
Keywords: bert nlp mxnet gluonnlp machine deep learning sentence encoding embedding
Platform: UNKNOWN
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Intended Audience :: Developers
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3 :: Only
Classifier: Programming Language :: Python :: 3.5
Classifier: Programming Language :: Python :: 3.6
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Information Analysis
Classifier: Operating System :: POSIX
Classifier: Operating System :: Unix
Classifier: Operating System :: MacOS
Description-Content-Type: text/markdown
Requires-Dist: mxnet (==1.3.0)
Requires-Dist: gluonnlp (==0.5.0)
Provides-Extra: gpu
Requires-Dist: mxnet-cu92 (==1.3.0) ; extra == 'gpu'

# Bert Embeddings

[![Build Status](https://travis-ci.org/imgarylai/bert_embedding.svg?branch=master)](https://travis-ci.org/imgarylai/bert_embedding)

![](https://img.shields.io/pypi/v/bert-embedding.svg?style=flat)

## Install

```
pip install bert-embedding
```
If you want to run on GPU machine, please install `mxnet-cu92`.

```
pip install mxnet-cu92
```

## Usage

```python
from bert_embedding import BertEmbedding

sentences = ["Hello World", "Token level embeddings from BERT model on mxnet and gluonnlp"]
bert = BertEmbedding()
result = bert.embedding(sentences)
```

