From 6d714b8f194bb5c1409b29e54531ce034f4efdaa Mon Sep 17 00:00:00 2001 From: malin10 Date: Mon, 24 Aug 2020 11:44:22 +0800 Subject: [PATCH] bug fix for python3 --- models/recall/word2vec/preprocess.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/recall/word2vec/preprocess.py b/models/recall/word2vec/preprocess.py index 679458e3..c55d22da 100755 --- a/models/recall/word2vec/preprocess.py +++ b/models/recall/word2vec/preprocess.py @@ -228,7 +228,7 @@ def data_split(args): contents.extend(f.readlines()) num = int(args.file_nums) - lines_per_file = len(contents) / num + lines_per_file = int(math.ceil(len(contents) / float(num))) print("contents: ", str(len(contents))) print("lines_per_file: ", str(lines_per_file)) -- GitLab