search

kevin-xuan / Graph-Flashback

[SIGKDD'2022] Graph-Flashback Network for Next Location Recommendation
https://dl.acm.org/doi/abs/10.1145/3534678.3539383
34 stars 7 forks source link

For training new dataset #5

Closed bb67ao closed 1 year ago

bb67ao commented 1 year ago

Could please make a readme if I want to train my dataset

kevin-xuan commented 1 year ago

Thanks for your interest in this research, I update the readme to guide training the model on your own datasets.

bb67ao commented 1 year ago

Thanks for your interest in this research, I update the readme to guide training the model on your own datasets.

Is there a more easier operation? The readme you updated is still a bit difficult for a novice, because I want to quickly test the effect on our dataset

kevin-xuan commented 1 year ago

OK, I just re-update the instructions for training your models on your own datasets, see section Training your dataset. Moreover, I upload the pure code of KGE model by this repository to training your model on new datasets, which cannot contain some files that are crucial for reproducing the original results in the research paper. See point 3 of section "Data preparation" for details.