Nempickaxe
/ word2vec.py
Last active
September 17, 2020 16:48
— forked from moustaki/Faster save-load for word2vec
Faster save-load for word2vec
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| import dbm, os | |
| import cPickle as pickle | |
| from gensim.models import Word2Vec | |
| import numpy as np | |
| def save_model(model, directory): | |
| model.init_sims() # making sure syn0norm is initialised | |
| if not os.path.exists(directory): | |
| os.makedirs(directory) | |
| # Saving indexes as DBM'ed dictionary |