From 4be3f7646dd072f5e11adbf5fab563dd270f72fb Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fl=C3=A1via=20Elias=20Rocha?= Date: Fri, 12 Apr 2019 16:57:29 -0300 Subject: [PATCH] Solving #50 --- glove/glove.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/glove/glove.py b/glove/glove.py index ec90ca3..2201bde 100644 --- a/glove/glove.py +++ b/glove/glove.py @@ -12,6 +12,7 @@ import numpy as np import scipy.sparse as sp import numbers +import sys from .glove_cython import fit_vectors, transform_paragraph @@ -163,8 +164,12 @@ 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) + if (sys.version_info.major < 3): + word_ids = np.array(cooccurrence.keys(), dtype=np.int32) + values = np.array(cooccurrence.values(), dtype=np.float64) + else: + 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