Skip to content

场景教程导览

本章节聚焦 Torch-RecHub 在不同推荐场景下的实战用法。文档中的代码默认基于仓库内置样本数据,建议在仓库根目录执行。

代码资源

  • 完整 Python 示例脚本:examples/
  • 文档内按步骤拆解的教程:docs/zh/tutorials/

教程列表

教程适用场景链接
CTR 预测排序 / 点击率预估CTR 预测教程
召回模型双塔召回 / 向量检索召回模型教程
多任务学习CTR/CVR 联合建模多任务教程

快速导航

CTR 预测(精排)

适合想快速跑通 WideDeep / DeepFM / DCN 的用户。

python
from torch_rechub.models.ranking import DeepFM
from torch_rechub.trainers import CTRTrainer

model = DeepFM(deep_features=deep_features, fm_features=fm_features, mlp_params={"dims": [256, 128]})
trainer = CTRTrainer(model, device="cuda:0")
trainer.fit(train_dl, val_dl)

查看完整教程 →

召回模型

适合想跑通 DSSM / YoutubeDNN / MIND 的双塔或多兴趣召回链路。

python
from torch_rechub.models.matching import DSSM
from torch_rechub.trainers import MatchTrainer

model = DSSM(user_features, item_features)
trainer = MatchTrainer(model, mode=0, device="cuda:0")
trainer.fit(train_dl)

查看完整教程 →

多任务学习

适合想了解 MMOE / PLE / ESMM 在 Ali-CCP 样本数据上的训练流程。

python
from torch_rechub.models.multi_task import MMOE
from torch_rechub.trainers import MTLTrainer

model = MMOE(features, task_types=["classification", "classification"], n_expert=8,
             expert_params={"dims": [16]}, tower_params_list=[{"dims": [8]}, {"dims": [8]}])
trainer = MTLTrainer(model, task_types=["classification", "classification"], device="cuda:0")
trainer.fit(train_dl, val_dl)

查看完整教程 →

模型使用示例

按模型分类的详细使用教程,每篇包含数据准备、模型配置、训练、评估和调优建议。

排序模型

模型说明链接
DeepFMFM + Deep 联合模型DeepFM 教程
Wide&Deep记忆 + 泛化联合模型Wide&Deep 教程
DCN / DCNv2显式特征交叉网络DCN 教程
DIN目标注意力序列模型DIN 教程
DIEN兴趣演化序列模型DIEN 教程
BSTTransformer 序列模型BST 教程

召回模型

模型说明链接
DSSM双塔语义匹配模型DSSM 教程
YoutubeDNNYouTube 深度召回模型YoutubeDNN 教程
MIND多兴趣胶囊网络召回MIND 教程

多任务模型

模型说明链接
MMOE多门控专家混合模型MMOE 教程
PLE渐进分层提取多任务模型PLE 教程

推荐的验证顺序

  1. 先跑 快速开始,确认环境、训练器、样本数据都可用。
  2. 再看 CTR 教程召回教程,理解完整数据流。
  3. 最后深入具体模型页,查看参数解释、调优建议和 ONNX / 可视化用法。