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 / 可视化用法。