tensorflow-DeepFM icon indicating copy to clipboard operation
tensorflow-DeepFM copied to clipboard

有没有实现sparse输入的deepfm

Open starkeisntein opened this issue 7 years ago • 2 comments
trafficstars

sparse_index = tf.placeholder(tf.int64, [None, 2]) sparse_ids = tf.placeholder(tf.int64, [None]) sparse_values = tf.placeholder(tf.float32, [None]) sparse_shape = tf.placeholder(tf.int64, [2]) ids = tf.SparseTensor(sparse_index, sparse_ids, sparse_shape) values = tf.SparseTensor(sparse_index, sparse_values, sparse_shape)

这样子的sparse输入

starkeisntein avatar Apr 11 '18 03:04 starkeisntein

这份代码写的比较早,没有使用 sparse tensor 的方式来做。我在另外一个 project XNN 有试过 sparse tensor & tf.nn.embedding_lookup_sparse,效率有些问题,可能是我实现的不对。

ChenglongChen avatar Apr 23 '18 17:04 ChenglongChen

不要用稀疏的方式,最好是设定固定输入长度,缺失值设置value为0,index任意即可。

sudongxiang avatar Nov 21 '18 11:11 sudongxiang