推荐系统小项目

最近在弄一个小项目，从头搭建一个推荐系统，包括 modeling 和 infra 的部分，两头堵；主要的目的一方面是想弄个 educational 的项目来帮助想入门推荐系统的小伙伴，另一方面也是想复现一下经典的算法和架构，练练手
repo 在这里: https://github.com/pyemma/Argo, 目前还在施工中，同时我会尝试把学术界和工业界最新的一些工作（e.g. LLM integration）也不断的更新进来；里面的一些实现可能不是最高效的，主要的目的还是想让实现更加的简单易懂（之后会加入优化篇章来专门针对不够搞笑的部分进行优化）
欢迎大家多提意见！

非著名农民 发表于 2025-4-30 22:25
俩问题，1推荐啥？2迷你版TorchRec？

1. 目前主要就是用公开的数据集，先从 MovieLen 入手，之后打算加入 Kuaishou 的那些数据集来做推荐；比如现在的简单的 recomendation engine 就是给个 user id 然后推荐 movie id
2. modeling 的部分算是简化版的 TorchRec 吧，但是会 include recommendation engine 的部分，这个我印象中 TorchRec 好像没有包括

更新一下最近的 update:
1. 针对 TransAct 有了一个简单版本的实现，这个是推荐系统领域一个很经典的用来做 user sequence modeling 的工作
2. 使用双塔模型来训练 item embedding purely from user collaborative signals, 这个是用来平替 Pinformer 的，TransAct 的一个针对输入的要求
3. 一个简单的 trainer 在 MovieLen small dataset 上实现 E2E 的 training

tommychang1983 发表于 2025-5-1 02:22
Alibaba的公开数据集也可以试试。实现哪些模型？wide and deep, DIN, deepFM这些我也很感兴趣可以一起研究 ...

嗯嗯，欢迎一起研究！

目前的计划是把一些经典模型先实现了，这个在 README 的 Roadmap 上面有写

现在已经实现了
1. 一个简化版本的 DIN，没有实现文章中新提出的 activation func
2. 双塔模型，这个是用来 export movie embedding
3. 一个非常非常非常简单的 recommendation engine, 之后会慢慢扩展，打算用开源工具来搭建
4. TransAct 模型还在施工中，目前只是实现了 TransAct 和 DCNv2 的 module, 还需要 E2E hook 在一起进行 training

俩问题，1推荐啥？2迷你版TorchRec？

Alibaba的公开数据集也可以试试。实现哪些模型？wide and deep, DIN, deepFM这些我也很感兴趣可以一起研究。用TensorFlow 还是PyTorch ？

多谢分享！

Youtube 上有一个很好的推荐系统相关的 tutorial.
搜 "ai alchemy recommendation system"

感谢分享

可以改成uv嘛

感谢分享！

lz你好，我目前在找machine learning engineer工作，之前没有相关经验，你觉得跟着你的这个项目做下来，会对找工作有帮助吗？谢谢