DeepFFM

渣大传送门：https://github.com/nzc

我的轮子有一段优化是借鉴渣大的部分源码，使得ffm的时间复杂度与fm相当，运行时间能缩短一半以上

环境

• Tensorflow 1.4
• CUDA 8.0
• CUDNN 6.0

注意，以上版本必须对应

其他版本不知道能不能运行，因为时间的缘故，我没来得及尝试其他版本

第一次接触Tensorflow的朋友可以直接按这个版本配置环境，具体安装方法百度即可

具体代码就是上面这三个了……

运行方法：

默认是多分类（包括二分类）任务，修改config里面的相关配置即可直接运行

config中包括数值特征、类别特征、不定长特征（如：{"兴趣":"篮球,音乐,旅游"}。一个记录项对应几个item的特征，注意分隔符为','）的处理方案。只需要把对应的columns写上去就可以了