Skip to content

DyGRec/DT4SR

BranchesTags

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits

data

data

 
 

README.md

README.md

 
 

main.py

main.py

 
 

metrics.py

metrics.py

 
 

model.py

model.py

 
 

modules.py

modules.py

 
 

requirements.txt

requirements.txt

 
 

sampler.py

sampler.py

 
 

util.py

util.py

 
 

Repository files navigation

DT4SR

Example running command for Beauty dataset: python main.py --dataset=Beauty --train_dir=default --lr=0.001 --hidden_units=32 --maxlen=50 --dropout_rate=0.5 --num_blocks=1 --l2_emb=0.0 --num_heads=2 --evalnegsample 1000

About

Code for CIKM 2021 best short paper nomination "Modeling Sequences as Distributions with Uncertainty for Sequential Recommendation" https://arxiv.org/abs/2106.06165

Topics

transformer recommendation sequential-models coldstart

Resources

Readme
Activity
Custom properties

Stars

15 stars

Watchers

2 watching

Forks

2 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages