Home

Awesome

VANext

Predciting Human Mobility via Variational Attention

Environment

Python 2.7 & Tensorflow

Details

  1. Source code: NYC_gated.py (the demo code is implemented for NYC dataset in our paper)

  2. AMSGrad.py: optimizer, you can choose Adam or AMSGrad.py

  3. 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.

Notes

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).