search

cheungdaven / DeepRec

An Open-source Toolkit for Deep Learning based Recommendation with Tensorflow.
GNU General Public License v3.0
1.14k stars 293 forks source link

model AFM parameter in build_network() #20

Open caowencai opened 5 years ago

caowencai commented 5 years ago
  1. 代码实现中num_hidden没用上
  2. 看起来是把特征嵌入的维度与注意力嵌入的向量维度都设置为num_factor=16了 我理解num_hidden是特征嵌入维度参数，num_factor是注意力嵌入维度参数吧？
caowencai commented 5 years ago

另外，请教一下valid_dim=3的作用， 默认应该是全排列，向量两两之间进行计算，所有特征都应该计算注意力。 valid_dim的作用就是人为筛选前面的部分特征用来计算注意力，是这样吧？

如果用途是筛选good features interaction，按照原作者以及你的实现方式，特征排序前面的valid_dim个特征就代表了注意力筛选的good features，这样的话特征排序也有必要？