tensornet icon indicating copy to clipboard operation
tensornet copied to clipboard

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

Open jackyhawk opened this issue 3 years ago • 1 comments

对于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')

jackyhawk avatar Jun 11 '21 03:06 jackyhawk

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

zhangqianjin avatar Apr 15 '22 14:04 zhangqianjin