Closed bokang-ugent closed 2 years ago
This is an implementation of the STAMP (Short-Term Attention/Memory Priority Model for Session-based Recommendation, Liu et. al., KDD 2018) model. The mode is also one of the earlier proposed neural model in the line of SBR research, often used as a baseline.
Changes
models/matching/stamp.py
. A new line that imports the model is added in models/maching/__init__.py
. The implementation follows directly from the paper, with a few features refer to the original TensorFlow implementation as they are not fully clear from the paper (e.g., the normalization of attention weights is not described in the paper).examples/matching/run_sbr.py
is updated to allow STAMP model to be evaluated on the benchmark datasets.examples/matching/README.md
.Results
Recall@20 | MRR@20 | |
---|---|---|
YOOCHOOSE1/64 | 0.6675 | 0.2859 |
YOOCHOOSE1/4 | 0.7079 | 0.3074 |
DIGINETICA | 0.5578 | 0.2303 |
Future work
The descriptions are added to examples/matching/README.md. Ranking metrics obtained by the NARM model is also added.