README.md

OpenCTR

目录

• 1. 简介
• 2. 特性
• 3. 准备模型与数据
• 4. 模型训练
• 5. 例程测试
• 6. ZOO

1. 简介

OpenCTR 是支持tensorflow/pytorch/onnx多种框架的点击率预测项目，涵盖了DNN/LR/FM/DeepFM/DCN等模型。

2. 特性

• 支持tensorflow/pytorch/onnx多种框架
• 支持DNN/LR/FM/DeepFM/DCN等模型

3. 准备模型与数据

您下载本项目即可，包含多个模型，以及Movielens1M数据集。

4. 模型训练

进入demo文件夹，运行各个XXX_demo.py文件即可。

5. 例程测试

• python、models、scripts三个文件夹是sophon-demo所需的文件夹
• 数据集要涵盖训练数据集和测试集

6. ZOO

给出一个ModelZOO例子，DeepFM。