
MiniOneRec
用最小代码跑通阿里OneRec推荐模型,快速上手统一召回排序
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 1959
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- Python 3.11(README 使用 conda 创建)
- conda 环境管理工具
- 已安装 git
- 4~8 张 A100/H100 GPU(复现所需)
- 能访问 GitHub 与依赖源
安装与启动步骤
-
1克隆仓库
从 GitHub 拉取 MiniOneRec 源码并进入项目目录,后续命令都在该目录下执行。
git clone https://github.com/AkaliKong/MiniOneRec.git cd MiniOneRec -
2创建conda环境
用 Python 3.11 新建名为 MiniOneRec 的独立环境并激活,避免与系统 Python 冲突。
conda create -n MiniOneRec python=3.11 -y conda activate MiniOneRec -
3安装依赖
按仓库提供的 requirements.txt 安装全部 Python 依赖,建议在网络稳定时执行。
pip install -r requirements.txt -
4运行评估脚本
执行仓库自带的 evaluate.sh 完成评测流程;README 说明使用官方提供的预训练 SID 可快速开始。
bash evaluate.sh
如何确认成功
评估脚本正常跑完且日志中 CC 指标为 0,说明约束解码生效、结果可信。
常见问题
Q:需要多少 GPU 才能复现?
A:README 明确说明复现需要 4~8 张 A100/H100 GPU,单卡很难完成完整复现。
Q:评估日志里 CC 指标非 0 是什么问题?
A:说明模型仍在大量生成无效 item,约束解码未成功,可能与 transformers 等依赖版本不一致有关,可参考 calc.py 排查。
Q:有没有预训练权重可以直接用?
A:有,官方在 Huggingface(kkknight/MiniOneRec)和 ModelScope 上发布了预训练模型,可配合官方提供的 Industrial/Office SID 快速开始。
Q:安装依赖失败怎么办?
A:先确认已激活 Python 3.11 的 MiniOneRec 环境,再检查网络与 pip 源;README 未提供其它安装方式,不建议自行替换包版本。
注意事项
- README 未给出数据下载与训练脚本的完整命令,训练/数据准备细节请以仓库最新文档为准。
- 不同依赖版本可能影响约束解码效果,务必关注评估日志中的 CC 指标。
- 项目包含 SID 构建、SFT 和面向推荐的强化学习三个环节,完整跑通耗时与显存开销都较大。
核心亮点
- 代码精简,完整复现OneRec核心流程,适合快速学习
- 提供示例数据和训练脚本,开箱即用,降低入门门槛
- 基于Python,易于扩展和集成到现有推荐系统
不足之处
- 文档和示例相对简略,复杂场景需自行探索
- 性能优化可能不如生产级推荐系统,需自行调优
适用场景
- 推荐系统研究者快速验证OneRec算法效果
- 工程师学习统一推荐模型的实现细节
- 作为基线模型,对比其他推荐算法的性能
替代项目
DeepCTR、FuxiCTR、RecBole
上一篇:LongWriter
下一篇:ModernBERT
同类项目推荐
awesome-fm4co
开源
一网打尽基础模型+组合优化最新论文,快速跟上研究前沿
Recent research papers about Foundation Models for Combinatorial Optimization
Otter
开源
让开源多模态模型听懂指令,看图说话更听话
Otter, a multi-modal model based on OpenFlamingo (open-sourced version of DeepMind's···
Chat-UniVi
开源
一个模型同时看懂图片和视频,统一视觉理解新方案
[CVPR 2024 Highlight] Chat-UniVi: Unified Visual Representation Empowers Large Langu···
LLaMA-Mesh
开源
用大模型直接生成 3D 网格,告别复杂建模流程
Unifying 3D Mesh Generation with Language Models
awesome-ai-models
开源
一张清单,快速掌握当下最值得关注的AI模型。
A curated list of top AI models and LLMs
LabVLA
开源
让机器人看懂实验手册,自动完成科研操作
LabVLA: Grounding Vision–Language–Action Models in Scientific Laboratories
nixtla
开源
几行代码调用预训练模型,搞定时间序列预测与异常检测。
TimeGPT-2.1: production ready pre-trained Time Series Foundation Model for forecast···
ProCyon
开源
融合多模态数据,精准预测蛋白质功能表型
ProCyon: A multimodal foundation model for protein phenotypes
Awesome-Interactive-World-Model
开源
一张地图看懂交互式世界模型研究,快速跟上前沿。
Interactive World Model papers organized by core research challenges.