tensorflow-DeepFM
tensorflow-DeepFM copied to clipboard
有没有实现sparse输入的deepfm
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输入
这份代码写的比较早,没有使用 sparse tensor 的方式来做。我在另外一个 project XNN 有试过 sparse tensor & tf.nn.embedding_lookup_sparse,效率有些问题,可能是我实现的不对。
不要用稀疏的方式,最好是设定固定输入长度,缺失值设置value为0,index任意即可。