AI论坛 AI论坛 beta

柯慕灵 (@ft19853) 在 基于pytorch的推荐算法框架 中发帖

📌 📄 内容摘要: 柯慕灵 (@ft19853) 在 基于pytorch的推荐算法框架 中发帖 向大家推荐自己负责维护的社区框架,目前700star了,代码风格采用机器学习时最爱的scikit-learn的风格。目前主要支持的功能和采用的模块如框架图所示: [image] 之前就在我前司用这个框架上线了双塔模型,嘎嘎好用。除去数据处理部分,10 行代码实现工业级推荐系统。30+ 主流模型开箱即用,支持一键 ONNX 部署,让你专注于业务而非工程。 使用手册在这里:torch-rechub 手册基本还算完善,期待有贡献者愿意补充。欢迎各位会用Torch,并且对推荐算法感兴趣的小伙伴加入我们~ 那么具体有哪些特性呢?在README都有写: 模块化设计: 易于添加新的模型、数据集和评估指标。 基于 PyTorch: 利用 PyTorch 的动态图和 GPU 加速能力。 丰富的模型库: 涵盖 30+ 经典和前沿推荐算法(召回、排序、多任务、生成式推荐等)。 标准化流程: 提供统...
────────── 链接信息 ──────────
🔗 论坛链接: linux.do
📎 访问地址: https://linux.do/t/topic/1500263/1
─────────────────────────────
📢 来源:LINUX DO

你好啊,陌生人!

我的朋友,看起来你是新来的,如果想参与到讨论中,点击下面的按钮!

每日一言

AI论坛

帖子数 659811
评论数 8888
用户数 88888
在线用户 8888