Qihoo360 / tensornet

Apache License 2.0
315 stars 72 forks source link

对于tensornet,是否支持一个slot下多个feature,需要对该slot下的多个feature 求sum/mean pooling? #52

Open jackyhawk opened 3 years ago

jackyhawk commented 3 years ago

对于tensornet,是否支持一个slot下多个feature,需要对该slot下的多个feature 求sum/mean pooling?

比如一个itemId,可能有两个tag,tag_科比,tag_库里,需要在tag这个slot下的多个tag,做下pooling,作为这个slot(field)的输入

一个用户的tag兴趣,可能有 tag_湖人,tag_科比,tag_勇士,tag_库里,需要把需要在tag兴趣这个slot下的多个tag兴趣,做下pooling,作为这个slot(field)的输入

还是说,直接用TensorFlow自带的即可? combiner_embed = tf.nn.embedding_lookup_sparse(self.embedding,sp_ids=idx, sp_weights=val, combiner='sum')

zhangqianjin commented 2 years ago

这个是支持的,我跟了下代码逻辑,一个slot下面的 feature的emb是通过tf自带的聚合,支持sum或者mean, 如果想取出一个slot下面的各个feature 的emb,需要额外的开发一小部分代码