Transformers4Rec

收录理由

如果你的产品需要预测用户接下来会点击或购买什么，Transformers4Rec 值得一试。它从 Hugging Face 借来 transformer 模型，把它们用在用户交互序列上——用户浏览过的每件商品、加入购物车的每个条目，都被当作句子里的一个 token 来处理。这种思路在访客匿名浏览时同样管用，因为模型只依赖当前会话中的行为记录。库本身基于 PyTorch 构建，你可以直接载入熟悉的 transformer 预训练权重，在自己的物品序列上训练，不必重新实现注意力机制。已经在用 Hugging Face 的研究人员和工程师会发现集成过程很顺手，库中还提供了可复用的构建模块，方便快速搭建基于会话的推荐系统原型。