import warnings
warnings.filterwarnings("ignore")
import gensim
import nltk
import numpy as np
CORPUS = [
'the sky is blue',
'sky is blue and sky is beautiful',
'the beatiful sky is so blue',
'I love blue cheese']
new_doc = ['loving this blue sky today']
TOKENIZED_CORPUS = [nltk.word_tokenize(sentence) for sentence in CORPUS]
tokenized_new_doc = [nltk.word_tokenize(sentence) for sentence in new_doc]
model = gensim.models.Word2Vec(TOKENIZED_CORPUS, size=10, window=10,min_count=2, sample=1e-3)
print(model['sky'])
print(model['blue']