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