提交 1e8a5f46 编写于 作者: F frankwhzhang

fix python3 xrange bug, gru4rec/tagspace

上级 9b93571e
......@@ -78,6 +78,6 @@ if __name__ == "__main__":
test_dir, "", batch_size=batch_size,
buffer_size=1000, word_freq_threshold=0, is_train=False)
for epoch in xrange(start_index, last_index + 1):
for epoch in range(start_index, last_index + 1):
epoch_path = model_dir + "/epoch_" + str(epoch)
infer(test_reader=test_reader, use_cuda=use_cuda, model_path=epoch_path)
......@@ -80,6 +80,6 @@ if __name__ == "__main__":
test_dir, "", vocab_tag_path, batch_size=1,
neg_size=0, buffer_size=1000, is_train=False)
for epoch in xrange(start_index, last_index + 1):
for epoch in range(start_index, last_index + 1):
epoch_path = model_dir + "/epoch_" + str(epoch)
infer(test_reader=test_reader, vocab_tag=vocab_tag, use_cuda=False, model_path=epoch_path)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册