This small library offers tools to make visualisation easier of both word embeddings as well as operations on them. It has support for spaCy prebuilt models as a first class citizen but also offers support for sense2vec. There's a convenient API to perform linear algebra as well as support for popular transformations like PCA/UMAP/etc.
from whatlies import EmbeddingSet from whatlies.language import SpacyLanguage lang = SpacyLanguage('en_core_web_md') words = ['cat', 'dog', 'fish', 'kitten', 'man', 'woman', 'king', 'queen', 'doctor', 'nurse'] emb = lang[words] emb.plot_interactive(x_axis='man', y_axis='woman')EXAMPLE