search

pmixer / SASRec.pytorch

PyTorch(1.6+) implementation of https://github.com/kang205/SASRec
Apache License 2.0
331 stars 90 forks source link

How to generate a list of items as the recommendation result? #29

Open ralgond opened 1 year ago

ralgond commented 1 year ago

thank you!

pmixer commented 1 year ago

@ralgond in short, you could sample some candidate items, scoring by the model, select topk items with higher scores, similar details shown in https://github.com/pmixer/SASRec.pytorch/blob/master/utils.py#L140 , still, the repo is for recsys research use, you may need to do some engineering work if hope to train a model and deploy it for online inference for production.