Torch-Rechub学习笔记-task1
这里写自定义目录标题
- Torch-Rechub学习笔记-task1
- 配置环境
- 跑tutorials中代码
- 学习心得
Torch-Rechub学习笔记-task1
以下以DatawhaleTorch—Rechub组队学习的Task笔记
[项目开源地址]: https://github.com/datawhalechina/torch-rechub
配置环境
本次学习并没有采用uv方式配置临时的虚拟环境,在本机上直接pip install 安装
跑tutorials中代码
在tutorials中实操了一下代码,以DSSM作为召回模型生成用户侧embedding和物品侧embedding最后使用milvus做向量召回返回k个物品。
学习心得
由于之前有参与过FunRec的学习,也使用deepctr复现过各个召回模型的代码。召回模型中的数据预处理以及生成embedding最后放入向量检索工具中计算并实现返回k个结果这一步骤在torch-rechub也大差不差。具体需要学习torch-rechub在实现排序模型和多目标任务中的步骤以及一些通用的点。后续会尝试使用torch-rechub做一个完整的项目。