gensimのembeddingをloadする

load_embedding_with_gensim.py

import linecache

from gensim.models import KeyedVectors


def load_embedding_with_gensim(embedding_name):
    '''
    Load embeddings with gensim.
    '''
    if embedding_name.endswith('bin'):
        binary = True
        no_header = False
    else:
        binary = False
        if linecache.getline(embedding_name, 1).split() == 2:
            no_header = False
        else:
            no_header = True
            
    embedding = KeyedVectors.load_word2vec_format(embedding_name, binary=binary, no_header=no_header)
	
    return embedding