SparrowRecSys icon indicating copy to clipboard operation
SparrowRecSys copied to clipboard

Published 20 hours ago • verified publisher icon wzhe06

Update DIEN.py

Open toyoululu opened this issue 2 years ago • 0 comments

1.修改了GRU中公式计算错误，去除了多余的线性层。 2.修改了positive_loss和negtive_loss定义，解决了损失值合并时可能出现负数的问题。 3.对用GPU训练可能会有问题，model.evaluate返回值接收不正确进行了修改

toyoululu avatar Dec 05 '21 12:12 toyoululu