Project word2vec vectors to 2D

wordproj.py

# Word list
Wl = ['man', 'woman', 'rich', 'poor', 'queen',
      'king', 'fisherman', 'teacher', 'actress', 'actor']
Wv = []
for i in range(len(Wl)):
    # Embeddings
    Wv.append(word2vec[Wl[i]])
# To-be basis
b1 = (Wv[1]-Wv[0])
b2 = (Wv[3]-Wv[2])
# Get pseudo-inverse matrix
W = np.array(Wv)
B = np.array([b1,b2])
Bi = np.linalg.pinv(B.T)
# Project all the words
Wp = np.matmul(Bi,W.T)
Wp = (Wp.T-[Wp[0,2],Wp[1,0]]).T