/poi2vec

POI2Vec: Geographical Latent Representation for Predicting Future Visitors

Primary LanguageJupyter Notebook

POI2Vec

POI2Vec: Geographical Latent Representation for Predicting Future Visitors Shanshan Feng, Gao Cong, Bo An, Yeow Meng Chee Proceedings of the Thirty-First AAAI Conference on Artificial Intelligence (AAAI-17)

How to preprocess

run the three scripts:

python preprocess.py
python preprocess2.py
python merge_data.py

How to train

python train.py --gpu <gpu-id>

or use default on cpu.