DIEN implementation is not aligned with DIEN paper for some reason

[andrei-kochin]

Hello,

Could you please kindly clarify the reason in 'u_t' calculation in DIEN model?

ai-matrix/macro_benchmark/DIEN/script/utils.py

Line 217 in f7e1d77

u = (1.0 - att_score) * u

According to DIEN paper it should be u_t =a_t*u_t and in your implementation it is u_t = (1.0 - a_t) * u_t

Looking forward for your reply.

Thank you!

[e-nugmanova]

Paper: https://arxiv.org/pdf/1809.03672.pdf

Formulas 11-12:

[andrei-kochin]

@rabbitwayne, @xulingjie @madeye @parallelo @ could you please help us to comment on that?