TiSASRec.pytorch
TiSASRec.pytorch copied to clipboard
pmixer
https://github.com/JiachengLi1995/TiSASRec in PyTorch
作者您好，我在ubuntu18.04环境下运行您的代码出现了failed loading state_dicts, pls check file path: ml-1m_default/TiSASRec.epoch=601.lr=0.001.layer=2.head=1.hidden=50.maxlen=200.pth这个错误，我是根据您给出的readme这个文件中的指令‘python main.py --dataset=ml-1m --train_dir=default --dropout_rate=0.2 --device=cpu --state_dict_path='ml-1m_default/TiSASRec.epoch=601.lr=0.001.layer=2.head=1.hidden=50.maxlen=200.pth' --inference_only=true --maxlen=200’直接运行的，只修改了cuda为cpu，就报了这个错误，pytorch版本是1.10.2，不知道哪里存在问题，还望指点一二，感谢
https://github.com/pmixer/TiSASRec.pytorch/blob/e87342ead6e90898234432f7d9b86e76695008bc/model.py#L104 thx for your nice work about implementing 'TiSASRec' model with pytorch. here is a issue: 'self.item_num+1' may be written in 'args.time_span+1', although with the condition of item_num > time_span...
Hi Guys, FYI, when you are trying changing the `num_heads` to be a number other than the default `num_heads=1`, pls be sure to make it a divisor of `hidden_units`, or...
Hi Guys, Weirdly, I forgot the activation function(ReLU) in former implementation which has been fixed in https://github.com/pmixer/TiSASRec.pytorch/commit/c4e6230b2568fb5099b60eeccdc2723734392c60, as we know, without activation functions, the whole network would just be a...
Great Job! When running the code I observed that the evaluate time is much larger than the training time is there any solution?
I want to know if the ml-1m dataset in this project and the ml-1m dataset in SASRec.pytorch are from the same source? I have tested both projects separately, it seems...
download the data you want from https://cseweb.ucsd.edu/~jmcauley/datasets/amazon/links.html, choose the 5-core format, and adjust the code below to process the raw data into the format required by the TiSASRec model. ```...