tensorflow-DeepFM

tensorflow-DeepFM copied to clipboard

模型auc

1

看着模型的auc 似乎挺低的，只有0.64左右，请问这个符合预期吗？ y label is: [0, 0, 0, 0, 0, 0, 0, 0, 0, 0] y pred is: [0.05798095 0.03533146 0.01641405 0.03143436 0.03862092 0.03267911 0.05392143 0.02281025 0.01579049 0.01842922] y label...

WoNiuHu

tensor flow version

2

Hi, do you have a recommendation of tensor flow version? Further, if you tell us the version of CUDA and cuDNN, it is highly appreciated.

KentaItakura

Angel0726

washujiang

换了一批数据，auc一直0.5，为什么？

15

VincentLiu123

这个项目的版本有吗 TensorFlow 用哪个版本啊

2

foreseez

换了一批其他数据，然后dfm.predict 结果为nan

8

换了一批其他数据，然后dfm.predict 结果为nan，求大神指点

balataTree

关于代码107行

3

deepFM.py 第107行为：self.y_deep = tf.nn.dropout(self.y_deep, self.dropout_keep_deep[0])，本人在实验中，删除该行，效果反而变好了很多。查阅资料后，发现dropout一般的对象是deep的中间节点，而该行是针对初始节点。 楼主可以自行实验，以加验证。 祝好~

sudongxiang

In DeepFm.py line 102, why the second order term's size is (None, K) rather than (None, )?

2

From the equation (2) in the paper, the second order term should be a scalar. However, in the code, it is computed as a vector. I think the line 102...

Wang-Yu-Qing

怎么将模型转换成pb文件呢？输出层和输入层是什么

1

KatouMegumiMoe