Skip to content

Latest commit

 

History

History
18 lines (13 loc) · 335 Bytes

README.md

File metadata and controls

18 lines (13 loc) · 335 Bytes

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