MiniOneRec

用最小代码跑通阿里OneRec推荐模型,快速上手统一召回排序

MiniOneRec 是一个精简复刻 OneRec 的开源项目,OneRec 是阿里巴巴推出的统一推荐模型,旨在解决推荐系统中多阶段任务(召回、排序、重排)割裂的问题。MiniOneRec 用 Python 实现了 OneRec 的核心架构,包括特征工程、模型训练和推理流程,让开发者能够快速理解和上手这一先进推荐算法。它提供了清晰的代码结构和示例数据,降低了学习门槛,同时保持了模型的可扩展性,方便研究者在此基础上进行二次开发和实验。项目解决了从论文到代码的落地难题,帮助推荐系统从业者快速验证和部署统一推荐模型。

开源 free 开源模型
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 1841
维护状态 较活跃
是否开源 是
定价模式 free

项目数据

分类开源模型
开发团队AkaliKong
所属国家
官网地址
定价模式free
价格说明开源项目,Apache-2.0许可证,可自由使用和部署,无付费版本。
访问状态
是否开源是
开源协议Apache-2.0
主要语言Python
技术栈/模型generative-ai,generative-recommendation,generative-recommenders,large-language-model,large-language-models,llm,recommendation,recommendation-system,recommender-system,scaling-laws
GitHub 星标★ 1841
30天Star增速
HF 下载量
上线时间2025-10-21 00:00:00
最近更新2026-09-23 00:00:00
维护状态较活跃
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数3

使用教程

难度:高级 约 60 分钟 部署方式:本地安装 4 步

环境要求

  • Python 3.11(README 使用 conda 创建)
  • conda 环境管理工具
  • 已安装 git
  • 4~8 张 A100/H100 GPU(复现所需)
  • 能访问 GitHub 与依赖源

安装与启动步骤

  1. 1克隆仓库

    从 GitHub 拉取 MiniOneRec 源码并进入项目目录,后续命令都在该目录下执行。

    git clone https://github.com/AkaliKong/MiniOneRec.git
    cd MiniOneRec
  2. 2创建conda环境

    用 Python 3.11 新建名为 MiniOneRec 的独立环境并激活,避免与系统 Python 冲突。

    conda create -n MiniOneRec python=3.11 -y
    conda activate MiniOneRec
  3. 3安装依赖

    按仓库提供的 requirements.txt 安装全部 Python 依赖,建议在网络稳定时执行。

    pip install -r requirements.txt
  4. 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

项目介绍

MiniOneRec 是开源模型领域的开源项目,由 AkaliKong 开发,2025 年首次发布。

在全站 13,014 个收录项目中,它的 GitHub 星标数(1,841)位列前 30%,在开源模型分类中处于中上游。

近 42 天,它的 GitHub 星标从 1,739 增加到 1,841,净增 102。

项目保持着较活跃的维护节奏,最近一次代码更新于 2026-09-23。Apache-2.0许可证,可自由使用和部署,无付费版本。

它主要面向的使用场景是:推荐系统研究者快速验证OneRec算法效果。同类可对比的替代方案包括 DeepCTR、FuxiCTR、RecBole。

上一篇:LongWriter

下一篇:ModernBERT

同类项目推荐

Machine-Learning 开源

跟着博客笔记,边看边跑机器学习代码

All content related to machine learning from my blog

★ 115 2026-08-09
CoreML-Models 开源

下载即用的苹果端 AI 模型库,免去转换烦恼

Core ML model zoo for iOS/macOS — PyTorch models converted to ready-to-use .mlpacka···

★ 1873 2026-08-10