Skip to content

ceo21ckim/SASRec-Pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SASRec-Pytorch

python main.py --dataset Beauty-Origin --use_wandb true --dr_rate 0.2

Origin hyper-parameter settings

Params Value
self-attention block 2
learning rate 0.001
dropout rate 0.2
maximum sequence length 200
Top@K 10
batch size 128

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages