DeepFM 二分类范例代码

前方干货预警：这可能是你能够找到的最容易理解和最容易跑通的DeepFM建模范例代码。

本范例演示使用 torchkeras中内置的DeepFM实现进行广告点击率预测。

公众号算法美食屋后台回复关键词：torchkeras，获取本文notebook源码和所用criteo_small数据集下载链接。

有关DeepFM模型的完整理论介绍和源码讲解，可以参考eat_pytorch_in_20_days中的讲解教程。

一，准备数据

二，定义模型

三，训练模型

四，评估模型

五，使用模型

六，保存模型

最佳模型权重已经保存在ckpt_path = 'checkpoint'位置了。