Predciting Human Mobility via Variational Attention
Python 2.7 & Tensorflow
-
Source code: NYC_gated.py (the demo code is implemented for NYC dataset in our paper)
-
AMSGrad.py: optimizer, you can choose Adam or AMSGrad.py
-
attention.py. It is an old version, if you want to use.
Btw, as to the poi embedding, you can use the causal embedding method described in our paper.
Pleas cite: Gao Q, Zhou F, Trajcevski G, Zhang K, Zhong T, Zhang F. Predicting human mobility via variational attention. InThe World Wide Web Conference 2019 May 13 (pp. 2750-2756).