We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a80e106 commit aac8937Copy full SHA for aac8937
python/deep_model.py
@@ -31,7 +31,7 @@ def concat(self, fields, sparse_ids, sparse_vals):
31
input_size = self.field_feature_dict.field2feanum[field_id] + 1
32
with tf.variable_scope("emb_"+str(field_id)):
33
embedding_variable = tf.Variable(tf.truncated_normal([input_size, self.embedding_size], stddev=0.05), name='emb' + str(field_id))
34
- embedding = tf.nn.embedding_lookup_sparse(embedding_variable, tf.cast(sparse_ids[i], tf.int32), sparse_vals[i], "mod", combiner="sum")
+ embedding = tf.nn.embedding_lookup_sparse(embedding_variable, sparse_ids[i], sparse_vals[i], "mod", combiner="sum")
35
emb.append(embedding)
36
self.embedding.append(embedding_variable)
37
0 commit comments