search

jc-LeeHub / Recommend-System-tf2.0

原理解析及代码实战，推荐算法也可以很简单 🔥 想要系统的学习推荐算法的小伙伴，欢迎 Star 或者 Fork 到自己仓库进行学习🚀 有任何疑问欢迎提 Issues，也可加文末的联系方式向我询问！
https://github.com/jc-LeeHub/
644 stars 216 forks source link

AFM的实现中特征交叉层只用了一部分的特征做特征交叉 #7

Open himiko14122 opened 2 years ago

himiko14122 commented 2 years ago

您好，我看了AFM的实现代码，在layer.py中第70行将输入划分为dense_inputs和sparse_inputs，但是在后续的interaction_layer中只使用了sparse_inputs的embedding来做特征交叉，dense_inputs在后续的流程中完全没有用到，这里不是应该将dense_inputs和sparse_inputs的embedding都用来做特征交叉吗？

jc-LeeHub commented 2 years ago

您好，我看了AFM的实现代码，在layer.py中第70行将输入划分为dense_inputs和sparse_inputs，但是在后续的interaction_layer中只使用了sparse_inputs的embedding来做特征交叉，dense_inputs在后续的流程中完全没有用到，这里不是应该将dense_inputs和sparse_inputs的embedding都用来做特征交叉吗？

要让dense_inputs也参与交叉的话，需要先做embedding，维度跟sparse_inputs的embedding保持一样，然后concat在一起做特征交叉即可。