画像の保存(fig.savefig( ))

word2vec後、t-sneで次元圧縮した画像を保存する方法。

図をplt.showで作成する前に画像ファイルを作っておく。

#ライブラリの読み込み
from sklearn.manifold import TSNE
import matplotlib.pyplot as plt
from gensim.models import word2vec

#モデルの読み出し
model = word2vec.Word2Vec.load('MODEL.model')

#x軸の決定
X = model[model.wv.vocab]

#次元圧縮の定義
tsne = TSNE(n_components=2, perplexity=100)
#次元圧縮の計算
X_tsne = tsne.fit_transform(X[:1000,:])

#保存先の図の準備
fig = plt.figure()

#図の作成
plt.scatter(X_tsne[:, 0], X_tsne[:, 1])
plt.show()

#図の保存(拡張子を指定することで、図のフォーマットを決めることができる)

fig.savefig("FILENAME.jpg")