Commit 44607bc6 authored by 赵威's avatar 赵威

add timer

parent 89d9b293
import os
import time
from gensim.models import word2vec
......@@ -28,4 +29,8 @@ def w2v_train(f_name, model_output_name):
if __name__ == "__main__":
begin_time = time.time()
w2v_train("dispose_problem.txt", "w2v_model")
print("total cost: {:.2f}mins".format((time.time() - begin_time) / 60))
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment