From 12b35586db7e54d5feab5a01a73aa6b9cb1e758b Mon Sep 17 00:00:00 2001 From: rsharmapty Date: Thu, 12 Apr 2018 10:16:31 +0530 Subject: [PATCH] compatible for fasttext --- scripts/convert_glove2numpy.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/scripts/convert_glove2numpy.py b/scripts/convert_glove2numpy.py index 21f7e05..642dab1 100644 --- a/scripts/convert_glove2numpy.py +++ b/scripts/convert_glove2numpy.py @@ -17,7 +17,7 @@ lines = list(open(glove_path)) embed_matrix = np.zeros((len(lines), args.dim), dtype='float32') for i, line in enumerate(lines): - parts = line.strip().split() + parts = line.rstrip().split(' ') word = parts[0] vec = np.array(parts[1:], dtype='float32') embed_matrix[i] = vec