OpenCTR 是支持tensorflow/pytorch/onnx多种框架的点击率预测项目,涵盖了DNN/LR/FM/DeepFM/DCN等模型。
- 支持tensorflow/pytorch/onnx多种框架
- 支持DNN/LR/FM/DeepFM/DCN等模型
您下载本项目即可,包含多个模型,以及Movielens1M数据集。
进入demo文件夹,运行各个XXX_demo.py文件即可。
- python、models、scripts三个文件夹是sophon-demo所需的文件夹
- 数据集要涵盖训练数据集和测试集
给出一个ModelZOO例子,DeepFM。