datawhalechina / torch-rechub

A Lighting Pytorch Framework for Recommendation Models, Easy-to-use and Easy-to-extend.
MIT License
400 stars 74 forks source link

Implementation of the STAMP model; Add SBR datasets description to examples/README #40

Closed bokang-ugent closed 2 years ago

bokang-ugent commented 2 years ago

The descriptions are added to examples/matching/README.md. Ranking metrics obtained by the NARM model is also added.

bokang-ugent commented 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

Results

Recall@20 MRR@20
YOOCHOOSE1/64 0.6675 0.2859
YOOCHOOSE1/4 0.7079 0.3074
DIGINETICA 0.5578 0.2303

Future work