diff --git a/fluid/PaddleRec/word2vec/reader.py b/fluid/PaddleRec/word2vec/reader.py index 14a29caa27c493911d4aa78b2c0f8509a048d061..ea3352ac28dc9685c6b0043170063003f95fc1d2 100644 --- a/fluid/PaddleRec/word2vec/reader.py +++ b/fluid/PaddleRec/word2vec/reader.py @@ -75,8 +75,8 @@ class Word2VecReader(object): self.id_frequencys = [ float(count) / word_all_count for count in self.id_counts_ ] - print("dict_size = " + str( - self.dict_size)) + " word_all_count = " + str(word_all_count) + print("dict_size = " + str(self.dict_size) + " word_all_count = " + str( + word_all_count)) self.random_generator = NumpyRandomInt(1, self.window_size_ + 1)