New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Update DIEN.py #59

Open

toyoululu wants to merge 1 commit into wzhe06:master from toyoululu:master

toyoululu commented Dec 5, 2021 •

edited

Loading

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


          Update DIEN.py

2029c82

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet