Code to return context for word on a current index, from a list of words

word2vec_1.py

def get_target(words, idx, window_size=5):
    ''' Get a list of words in a window around an index. 
        words -- list of words in text
        idx -- index of the center word
        window_size -- the window size to define context
    '''
    
    # implement this function
    R = np.random.randint(1, window_size+1)
    start = idx - R if (idx - R) > 0 else 0
    stop = idx + R
    target_words = words[start:idx] + words[idx+1:stop+1]
    
    return list(target_words)