From 4e9cb2e9d499ef9aa83a16e24f9b0a667d88882f Mon Sep 17 00:00:00 2001 From: zeke Date: Fri, 10 Nov 2017 12:05:55 -0800 Subject: [PATCH] fix to numpy not accepting dict keys/values iterators --- glove/glove.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/glove/glove.py b/glove/glove.py index ec90ca3..8a66a87 100644 --- a/glove/glove.py +++ b/glove/glove.py @@ -163,8 +163,8 @@ def transform_paragraph(self, paragraph, epochs=50, ignore_missing=False): random_state = check_random_state(self.random_state) - word_ids = np.array(cooccurrence.keys(), dtype=np.int32) - values = np.array(cooccurrence.values(), dtype=np.float64) + word_ids = np.array(list(cooccurrence.keys()), dtype=np.int32) + values = np.array(list(cooccurrence.values()), dtype=np.float64) shuffle_indices = np.arange(len(word_ids), dtype=np.int32) # Initialize the vector to mean of constituent word vectors