where can I find the topk location in the model(.pth)?

yingtaoluo / Spatial-Temporal-Attention-Network-for-POI-Recommendation

Codes for a WWW'21 Paper. POI recommender system for location/trajectory prediction.

https://doi.org/10.1145/3442381.3449998

164 stars 39 forks source link

Closed lixin716 closed 1 year ago

lixin716 commented 2 years ago

hello, I am reading your code,nice work! and I do not know how to get the topk locations in the model.

And I also want to know the meaning of the state_dict,like MultiEmbed.emb_t.weight,MultiEmbed.emb_su.weight,Attn.emb_loc.weight and so on.

Thanks,

lixin716 commented 2 years ago

@yingtaoluo

yingtaoluo commented 2 years ago

These are pytorch specified languages. You can search for topk function of numpy or torch. weight here means learnable parameters...