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.