Commit 70208d2e authored by 赵威's avatar 赵威

get first 15 lines

parent 5acbb892
......@@ -53,6 +53,7 @@ def get_tractate_vector(sentence_lst, all_keywords_set, model):
if s in all_keywords_set:
tmp_lst.append(s)
word_list.append(tmp_lst)
word_list = word_list[:15]
count = 0
res = {}
for lst in word_list:
......@@ -80,7 +81,6 @@ def save_tractate_vector_to_redis(all_keywords_set, model):
vec = get_tractate_vector(sentences, all_keywords_set, model)
redis_key = "rims:tractate:sentense:vector:" + str(id)
if vec:
vec = vec[:15]
print(count, id, len(vec))
redis_client5.set(redis_key, json.dumps(vec))
# TODO remove
......
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