DeepFM 二分类范例代码

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

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

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

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

一,准备数据

二,定义模型

picture.image

picture.image

三,训练模型

picture.image

picture.image

四,评估模型

picture.image

五,使用模型

picture.image

六,保存模型

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

公众号 算法美食屋 后台回复关键词: torchkeras,获取本文notebook源码和更多有趣范例。

picture.image

picture.image

picture.image

picture.image

0
0
0
0
评论
未登录
暂无评论