前方干货预警:这可能是你能够找到的最容易理解和最容易跑通的DeepFM建模范例代码。
本范例演示使用 torchkeras中内置的DeepFM实现进行广告点击率预测。
公众号 算法美食屋 后台回复关键词: torchkeras ,获取本文notebook源码和所用criteo_small数据集下载链接。
有关DeepFM模型的完整理论介绍和源码讲解,可以参考eat_pytorch_in_20_days中的讲解教程。
一,准备数据
二,定义模型
三,训练模型
四,评估模型
五,使用模型
六,保存模型
最佳模型权重已经保存在ckpt_path = 'checkpoint'位置了。
公众号 算法美食屋 后台回复关键词: torchkeras,获取本文notebook源码和更多有趣范例。