A little python code to show how to get similarity between word embeddings returned from the Rosette API's new /text-embedding endpoint.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.
README.md Fixed missing space in heading Mar 16, 2017
test_embeddings.py First commit of two python files to call the /text-embedding endpoint… Sep 14, 2016


Rosette API Text Embeddings Sample Code

This is a little python code to show how to calculate the similarity between words by computing the cosine similarity (using numpy) between the words' embeddings, returned from the Rosette API's new /text-embedding endpoint. The call to the API uses the 1.3 version of the python binding, so be sure to install that package via $ pip install rosette-api or --upgrade via pip to get the latest.

To try it out

  1. Clone the repo and open the files in your favorite text editor/python IDE.
  2. In cosine_similarity.py, replace the user_key parameter's value [your key here] with your Rosette API key and save.
  3. Run test_embeddings.py via your python IDE or command line: $ python test_embeddings.py

Customize for your data

Try editing test_embeddings.py to compare words OR longer text you might be interested in to see how their embeddings compare. And if you find anything interesting, let us know! Find us at support.rosette.com or support@rosette.com.