Skip to content

Latest commit

 

History

History
24 lines (17 loc) · 1.08 KB

README.md

File metadata and controls

24 lines (17 loc) · 1.08 KB

self-attention-recommendation

Keras + Tensorflow 2.0 implementation of Self-Attentive Sequential Recommendation. In Proceedings of IEEE International Conference on Data Mining (ICDM'18)

Setup

You have to set a environment variable, before you execute python scripts.

export TF_KERAS=1

Related Repositories

I use some repositories to implement the self-attention-recommendation.

Repository Description
kang205/SASRec Official implementation
CyberZHG/keras-multi-head Keras implementation of multi head self attention
CyberZHG/keras-pos-embd Keras implementation of position embedding
CyberZHG/keras-position-wise-feed-forward Keras implementation of point wise feed forward
CyberZHG/keras-layer-normalization Keras implementation of layer normalization