Spatial-Temporal-Attention-Network-for-POI-Recommendation icon indicating copy to clipboard operation
Spatial-Temporal-Attention-Network-for-POI-Recommendation copied to clipboard

Published 20 hours ago • verified publisher icon yingtaoluo

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

Open lixin716 opened this issue 2 years ago • 2 comments

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 avatar May 22 '22 12:05 lixin716

@yingtaoluo

lixin716 avatar May 22 '22 12:05 lixin716

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

yingtaoluo avatar Jun 05 '22 15:06 yingtaoluo