open file

63796e00 · 赵威 · 45f0e232 · 63796e00
Commit 63796e00 authored Oct 14, 2020 by 赵威
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

word_to_vec.py word_vector/word_to_vec.py +2 -1

No files found.
--- a/word_vector/word_to_vec.py
+++ b/word_vector/word_to_vec.py
@@ -14,7 +14,8 @@ class W2vSentences:
        self.f_name = f_name

    def __iter__(self):
-        for line in open(self.f_name, mode="r", encoding="utf-8", errors="ignore"):
+        with open(self.f_name, mode="r", encoding="utf-8", errors="ignore") as f:
+            for line in f.lines():
                yield line.split()