修改个人主页，优化推荐系统
Change-Id: I533e36dc891b1b60fcb1e4a71b522b3ba9d77984
diff --git a/recommend/utils/data_generator.py b/recommend/utils/data_generator.py
index e50b4e9..d3ef9cf 100644
--- a/recommend/utils/data_generator.py
+++ b/recommend/utils/data_generator.py
@@ -67,8 +67,6 @@
sid = seed2idx[seed_id]
user_items[uid].append(sid)
user_times[uid].append(ts)
- print(user_items)
- print(user_times)
with open(output_path, "w", encoding="utf-8") as f:
for uid in sorted(user_items.keys()):
items = " ".join(str(item) for item in user_items[uid])
@@ -76,12 +74,8 @@
f.write(f"{uid}\t{items}\t{times}\n")
-def main():
+def build_user_seed_graph():
download_rows, favorite_rows = fetch_data()
records, user_set, seed_set = process_records(download_rows, favorite_rows)
user2idx, seed2idx = build_id_maps(user_set, seed_set)
group_and_write(records, user2idx, seed2idx)
-
-
-if __name__ == "__main__":
- main()
\ No newline at end of file