PaddleRec 开源项目使用教程

PaddleRec 开源项目使用教程

PaddleRecRecommendation Algorithm大规模推荐算法库，包含推荐系统经典及最新算法LR、Wide&Deep、DSSM、TDM、MIND、Word2Vec、Bert4Rec、DeepWalk、SSR、AITM，DSIN，SIGN，IPREC、GRU4Rec、Youtube_dnn、NCF、GNN、FM、FFM、DeepFM、DCN、DIN、DIEN、DLRM、MMOE、PLE、ESMM、ESCMM, MAML、xDeepFM、DeepFEFM、NFM、AFM、RALM、DMR、GateNet、NAML、DIFM、Deep Crossing、PNN、BST、AutoInt、FGCNN、FLEN、Fibinet、ListWise、DeepRec、ENSFM，TiSAS，AutoFIS等，包含经典推荐系统数据集criteo 、movielens等项目地址:https://gitcode.com/gh_mirrors/pa/PaddleRec

1. 项目的目录结构及介绍

PaddleRec 项目的目录结构如下：

PaddleRec/
├── models/
│   ├── rank/
│   │   ├── dnn/
│   │   │   ├── config.yaml
│   │   │   ├── net.py
│   │   │   ├── reader.py
│   │   │   └── trainer.py
│   ├── recall/
│   ├── multi-task/
│   └── ...
├── tools/
│   ├── trainer.py
│   ├── static_trainer.py
│   └── ...
├── README.md
├── LICENSE
└── ...

目录结构介绍

• models/: 包含各种推荐算法的模型实现，如排序模型（rank）、召回模型（recall）、多任务模型（multi-task）等。
• tools/: 包含训练工具脚本，如动态图训练脚本（trainer.py）和静态图训练脚本（static_trainer.py）。
• README.md: 项目介绍文档。
• LICENSE: 项目许可证文件。

2. 项目的启动文件介绍

PaddleRec 项目的启动文件主要位于 tools/ 目录下，常用的启动文件包括：

• tools/trainer.py: 用于动态图训练的启动脚本。
• tools/static_trainer.py: 用于静态图训练的启动脚本。

启动文件介绍

• trainer.py: 该脚本用于启动动态图训练，可以通过命令行参数指定模型配置文件路径，例如：

  python -u tools/trainer.py -m models/rank/dnn/config.yaml
  

• static_trainer.py: 该脚本用于启动静态图训练，同样可以通过命令行参数指定模型配置文件路径，例如：

  python -u tools/static_trainer.py -m models/rank/dnn/config.yaml
  

3. 项目的配置文件介绍

PaddleRec 项目的配置文件通常位于各个模型目录下，以 config.yaml 命名。以下是一个典型的配置文件示例：

# models/rank/dnn/config.yaml

# 训练参数
train_params:
  use_gpu: True
  epoch: 10
  batch_size: 128

# 模型参数
model_params:
  embedding_size: 16
  hidden_units: [512, 256, 128]

# 数据参数
data_params:
  train_data_path: "data/train"
  test_data_path: "data/test"

配置文件介绍

• train_params: 包含训练相关的参数，如是否使用 GPU、训练轮数（epoch）、批大小（batch_size）等。
• model_params: 包含模型相关的参数，如嵌入层大小（embedding_size）、隐藏层单元数（hidden_units）等。
• data_params: 包含数据路径相关的参数，如训练数据路径（train_data_path）、测试数据路径（test_data_path）等。

通过修改配置文件中的参数，可以灵活地调整训练和模型的配置。

PaddleRecRecommendation Algorithm大规模推荐算法库，包含推荐系统经典及最新算法LR、Wide&Deep、DSSM、TDM、MIND、Word2Vec、Bert4Rec、DeepWalk、SSR、AITM，DSIN，SIGN，IPREC、GRU4Rec、Youtube_dnn、NCF、GNN、FM、FFM、DeepFM、DCN、DIN、DIEN、DLRM、MMOE、PLE、ESMM、ESCMM, MAML、xDeepFM、DeepFEFM、NFM、AFM、RALM、DMR、GateNet、NAML、DIFM、Deep Crossing、PNN、BST、AutoInt、FGCNN、FLEN、Fibinet、ListWise、DeepRec、ENSFM，TiSAS，AutoFIS等，包含经典推荐系统数据集criteo 、movielens等项目地址:https://gitcode.com/gh_mirrors/pa/PaddleRec