Data

Tools

Indexes

Applications

Experiments

Open Source Science

cs431_srecbs

https://github.com/duckviet/cs431_srecbs

Science Score: 44.0%

This score indicates how likely this project is to be science-related based on various indicators:

  • ✓
    CITATION.cff file
    Found CITATION.cff file
  • ✓
    codemeta.json file
    Found codemeta.json file
  • ✓
    .zenodo.json file
    Found .zenodo.json file
  • ○
    DOI references
  • ○
    Academic publication links
  • ○
    Academic email domains
  • ○
    Institutional organization owner
  • ○
    JOSS paper metadata
  • ○
    Scientific vocabulary similarity
    Low similarity (5.4%) to scientific vocabulary
Last synced: 7 months ago · JSON representation ·

Repository

Basic Info
  • Host: GitHub
  • Owner: duckviet
  • Language: Python
  • Default Branch: master
  • Size: 17.7 MB
Statistics
  • Stars: 0
  • Watchers: 1
  • Forks: 0
  • Open Issues: 0
  • Releases: 0
Created 11 months ago · Last pushed 11 months ago
Metadata Files
Readme Citation

README.md

modified based on SASRec.pytorch, PyTorch(1.6+) implementation of SASRec for simplicity, fixed issues like positional embedding usage etc. (making it harder to overfit, except for that, in recsys, personalization=overfitting sometimes)

@software{Huang_SASRec_pytorch, author = {Huang, Zan}, title = {PyTorch implementation for SASRec}, url = {https://github.com/pmixer/SASRec.pytorch}, year={2020} }

to train:

python main.py --dataset=ml-1m --train_dir=default --maxlen=200 --dropout_rate=0.2 --device=cuda

just inference:

``` python main.py --device=cuda --dataset=ml-1m --traindir=default --statedictpath='ml-1mdefault/SASRec.epoch=1000.lr=0.001.layer=2.head=1.hidden=50.maxlen=200.pth' --inference_only=true --maxlen=200

```

output for each run would be slightly random, as negative samples are randomly sampled, here's my output for two consecutive runs:

1st run - test (NDCG@10: 0.5897, HR@10: 0.8190) 2nd run - test (NDCG@10: 0.5918, HR@10: 0.8225)

Check paper author's repo

@inproceedings{kang2018self, title={Self-attentive sequential recommendation}, author={Kang, Wang-Cheng and McAuley, Julian}, booktitle={2018 IEEE International Conference on Data Mining (ICDM)}, pages={197--206}, year={2018}, organization={IEEE} }

Owner

  • Login: duckviet
  • Kind: user

Citation (CITATION.cff) 

# This CITATION.cff file was generated with cffinit.
# Visit https://bit.ly/cffinit to generate yours today!

cff-version: 1.2.0
title: SASRec.pytorch
message: the pytorch implementation of SASRec model
type: software
authors:
  - given-names: Zan
    family-names: Huang
    email: dreaming_hz@hotmail.com
    affiliation: NVIDIA
    orcid: 'https://orcid.org/0000-0003-3585-9467'
identifiers:
  - type: url
    value: 'https://github.com/pmixer/SASRec.pytorch'
    description: SASRec.pytorch github repository
repository-code: 'https://github.com/pmixer/SASRec.pytorch'

GitHub Events

Total
  • Push event: 4
  • Create event: 1
Last Year
  • Push event: 4
  • Create event: 1