Jovian
Sign In
Learn practical skills, build real-world projects, and advance your career
import nltk
from gensim.models import word2vec
from gensim.models.word2vec import Word2Vec
import numpy as np
import matplotlib.pyplot as plt
import os

Creating our sentences to train the word vectors

TextCorpus  = ["I like ISME",
               "ISME has a good MBA program",
               "ISME has good faculty",
               "Aiswarya is that good faculty",
               "I like ML"
]
text_tokens = [sent.split() for sent in TextCorpus]
text_tokens[:2]
[['I', 'like', 'ISME'], ['ISME', 'has', 'a', 'good', 'MBA', 'program']]