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

update ids

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