有关ripplenet损失函数

[tracy221]

您好，请问下ripplenet相关代码中，损失函数的基础部分交叉熵损失函数的代码在model.py的哪一块有具体的体现呢，代码我看的不是很明白。 另外，我跑其他推荐模型的时候都会在通用的train.py文件中报如下错误，想问下您知不知道可能的原因 line 79, in on_epoch_end log(epoch, logs['loss'], logs['AUC'], logs['Precision'], logs['Recall'], KeyError: 'AUC' 感谢并期待您的回复！

[DabeyHolmes]

您好，请问下ripplenet相关代码中，损失函数的基础部分交叉熵损失函数的代码在model.py的哪一块有具体的体现呢，代码我看的不是很明白。 另外，我跑其他推荐模型的时候都会在通用的train.py文件中报如下错误，想问下您知不知道可能的原因 line 79, in on_epoch_end log(epoch, logs['loss'], logs['AUC'], logs['Precision'], logs['Recall'], KeyError: 'AUC' 感谢并期待您的回复！

请问用的是什么环境，我用的python3.8 tf2.4.0跑ripplenet会出错

[tracy221]

您好，请问下ripplenet相关代码中，损失函数的基础部分交叉熵损失函数的代码在model.py的哪一块有具体的体现呢，代码我看的不是很明白。 另外，我跑其他推荐模型的时候都会在通用的train.py文件中报如下错误，想问下您知不知道可能的原因 line 79, in on_epoch_end log(epoch, logs['loss'], logs['AUC'], logs['Precision'], logs['Recall'], KeyError: 'AUC' 感谢并期待您的回复！

请问用的是什么环境，我用的python3.8 tf2.4.0跑ripplenet会出错

python 3.7 tensorflow 2.3.0

[tracy221]

还想问下损失函数的部分，知识图谱嵌入损失的部分我不太理解 kge_loss = 0 # 知识图谱嵌入损失项 for hop in range(hop_size): h_expanded = tf.expand_dims(h[hop], axis=2) # batch, ripple_size, 1, dim t_expanded = tf.expand_dims(t[hop], axis=3) # batch, ripple_size, dim, 1 hRt = tf.squeeze(h_expanded @ r[hop] @ t_expanded) # batch, ripple_size kge_loss += tf.reduce_mean(tf.sigmoid(hRt)) 为什么要对相乘后的hRt外面套一个sigmoid函数再求平均呢

[SSSxCCC]

使用最新版本的代码和readme中要求的版本就不会报KeyError: 'AUC'这个错误了。 损失函数照着算法原作者的github写的，我也不是很清楚为什么要这样算。