Skip to content

Instantly share code, notes, and snippets.

@Yuktha-Majella
Created August 7, 2021 16:15
Creating a Word2Vec model in Gensim
from gensim.models.word2vec import Word2Vec
from multiprocessing import cpu_count
import gensim.downloader as api
dataset = api.load("text8")
words = [d for d in dataset]
data1 = words[:1000]
w2v_model = Word2Vec(data1, min_count = 0, workers=cpu_count())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment