Merge pull request #1 from mino-park7/issue137

fix lovit#137 contexts matrix dtype error
mino-park7 · Oct 31, 2021 · a37de90 · a37de90
2 parents 503eaee + eba0d4a
commit a37de90
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/soynlp/vectorizer/_word_context.py b/soynlp/vectorizer/_word_context.py
@@ -119,4 +119,4 @@ def _encode_as_matrix(word2contexts, vocab2idx, verbose):
         print('  - (word, context) matrix was constructed. shape = {}{}'.format(
             x.shape, ' '*20))
 
-    return x
+    return x.astype(float)