Skip to content

Commit aac8937

Browse files
committed
fix
1 parent a80e106 commit aac8937

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/deep_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def concat(self, fields, sparse_ids, sparse_vals):
3131
input_size = self.field_feature_dict.field2feanum[field_id] + 1
3232
with tf.variable_scope("emb_"+str(field_id)):
3333
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")
34+
embedding = tf.nn.embedding_lookup_sparse(embedding_variable, sparse_ids[i], sparse_vals[i], "mod", combiner="sum")
3535
emb.append(embedding)
3636
self.embedding.append(embedding_variable)
3737

0 commit comments

Comments
 (0)