KGCN算法在Book-Crossing数据集上的结果问题

SSSxCCC / Recommender-System

A developing recommender system in tensorflow2. Algorithm: UserCF, ItemCF, LFM, SLIM, GMF, MLP, NeuMF, FM, DeepFM, MKR, RippleNet, KGCN and so on.

MIT License

410 stars 93 forks source link

Open hguomin opened 4 years ago

hguomin commented 4 years ago

Hi,

KGCN算法在Book-Crossing数据集上跑的时候，将negative_sample_threshold设置为大于0的值时，代码会抛出如下错误：

将negative_sample_threshold设置为0时，代码可以跑下去，但是结果的值低的可怜：

请问如果要正确跑出Book-Crossing的结果，参数应该如何设置？谢谢。

flyking1994 commented 3 years ago

我也遇到了这样的问题，很奇怪，RippleNet的数据预处理部分是一样的但是没有问题。请问您解决了吗

TONKIA commented 3 years ago

代码里面少个判断，按照下图改一下就可以了