word2index.py

from operator import itemgetter
from collections import OrderedDict

def word2index(model):
    """Extract word2index mapping from Word2Vec model.
    """
    word2index = OrderedDict((v, k) for k, v in sorted(model.index2word, key=itemgetter(1)))
    return word2index