code: def similarity(word1: str, word2: str) -> float:
      
      return _MODEL.similarity(word1, word2)
docs: Get cosine similarity between two words.
      If a word is not in the vocabulary, KeyError will be raised.
  
      :param string word1: first word
      :param string word2: second word
      :return: the cosine similarity between the two word vectors