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
Code for CIKM 2021 best short paper nomination "Modeling Sequences as Distributions with Uncertainty for Sequential Recommendation" https://arxiv.org/abs/2106.06165
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
Code for CIKM 2021 best short paper nomination "Modeling Sequences as Distributions with Uncertainty for Sequential Recommendation" https://arxiv.org/abs/2106.06165