Commit 1a4e0fff authored by 赵威's avatar 赵威

update ids

parent 21aaf34f
......@@ -9,8 +9,8 @@ model_dir = os.path.join(base_dir, "_models")
data_dir = os.path.join(base_dir, "_data")
if __name__ == "__main__":
id1 = ["366825", "203194"] # 下颚角
id2 = ["87874", "84198"] # 双眼皮
id1 = ["202647", "386735"] # 下颚角 5/1252645
id2 = ["87874", "84198"] # 双眼皮 95
with open(os.path.join(data_dir, "click_tractate_ids.csv"), "r") as f:
data = f.readlines()
......
......@@ -90,7 +90,7 @@ def save_clicked_tractate_ids_item2vec():
data = f.readlines()
for i in data:
tmp = i.split("|")
# device_id = tmp[0]
# app_session_id = tmp[0]
ids = tmp[1].rstrip("\n").split(",")
click_ids.append(ids)
model = Word2Vec(click_ids, hs=0, min_count=3, workers=multiprocessing.cpu_count(), iter=10)
......
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