word2vec词向量可视化

import gensim
import gensim.downloader as api
model = api.load('glove-wiki-gigaword-50')

import seaborn as sns
import matplotlib.pyplot as plt
import numpy as np
###########################单个king
plt.figure(figsize=(15, 1))
sns.heatmap([model["king"]],
            xticklabels=False,
            yticklabels=False,
            cbar=False,
            vmin=-2,
            vmax=2,
            linewidths=0.7)
plt.show()

######################不同单词对比
plt.figure(figsize=(15, 4))
sns.heatmap([
    model["king"],
    model["man"],
    model["woman"],
    model["king"] - model["man"] + model["woman"],
    model["queen"],
],
            cbar=True,
            xticklabels=False,
            yticklabels=False,
            linewidths=1)
plt.show()