From 6d012b2df9a9d8d0b190ed788133351fa86343b9 Mon Sep 17 00:00:00 2001 From: Rupsa Saha Date: Fri, 15 Jun 2018 12:14:24 +0530 Subject: [PATCH] added function return_word_vector(self,word) to Glove class to retrieve vector from loaded model --- glove/glove.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/glove/glove.py b/glove/glove.py index ec90ca3..7068bd4 100644 --- a/glove/glove.py +++ b/glove/glove.py @@ -307,3 +307,21 @@ def most_similar_paragraph(self, paragraph, number=5, **kwargs): paragraph_vector = self.transform_paragraph(paragraph, **kwargs) return self._similarity_query(paragraph_vector, number) + + def return_word_vector(self, word): + """ + returns glove vector corresponding to word + """ + + if self.word_vectors is None: + raise Exception('Model must be fit before querying') + + if self.dictionary is None: + raise Exception('No word dictionary supplied') + + try: + word_idx = self.dictionary[word] + except KeyError: + raise Exception('Word not in dictionary') + + return self.word_vectors[word_idx]