GeoMAN - GitLite

KO | EN

gitlite — search

#javascript #python #react #typescript #go #machine-learning #rust #linux #docker #macos #java #windows #cli #vue

GeoMAN

(★ 144)

Tensorflow Implementation of GeoMAN, IJCAI-18

File Explorer

Download Latest Version (.zip)

- - scaler-0.pkl
  - scaler-1.pkl
  - scaler-10.pkl
  - scaler-11.pkl
  - scaler-12.pkl
  - scaler-13.pkl
  - scaler-14.pkl
  - scaler-15.pkl
  - scaler-16.pkl
  - scaler-17.pkl
  - scaler-18.pkl
  - scaler-2.pkl
  - scaler-3.pkl
  - scaler-4.pkl
  - scaler-5.pkl
  - scaler-6.pkl
  - scaler-7.pkl
  - scaler-8.pkl
  - scaler-9.pkl
- README.txt
- AirQualityGeoMan.json
- framework.png
- log-folder-format.txt
- ijcai-2018-GeoMAN.pdf
- GeoMAN-12-6-eval-decoder_gts.npy
- GeoMAN-12-6-eval-external_inputs.npy
- GeoMAN-12-6-eval-global_attn_state_indics.npy
- GeoMAN-12-6-eval-global_input_indics.npy
- GeoMAN-12-6-eval-local_inputs.npy
- GeoMAN-12-6-global_attn_state.npy
- GeoMAN-12-6-global_inputs.npy
- GeoMAN-12-6-test-decoder_gts.npy
- GeoMAN-12-6-test-external_inputs.npy
- GeoMAN-12-6-test-global_attn_state_indics.npy
- GeoMAN-12-6-test-global_input_indics.npy
- GeoMAN-12-6-test-local_inputs.npy
- GeoMAN-12-6-train-decoder_gts.npy
- GeoMAN-12-6-train-external_inputs.npy
- GeoMAN-12-6-train-global_attn_state_indics.npy
- GeoMAN-12-6-train-global_input_indics.npy
- GeoMAN-12-6-train-local_inputs.npy
- README.txt
.gitignore
__init__.py
base_model.py
GeoMAN.py
LICENSE
README.md
test_model.ipynb
test_model.py
train_model.ipynb
train_model.py
utils.py

// repository documentation

Was this content helpful?

★ 0 (0 ratings)

**GeoMAN**
======
[GeoMAN](https://github.com/yoshall/GeoMAN): Multi-level Attention Networks for Geo-sensory Time Series Prediction.

*An easy implement of GeoMAN using TensorFlow, tested on `CentOS 7` and `Windows Server 2012 R2`.*

->[Pytorch version](https://github.com/xchadesi/GeoMAN)

## Paper 
[Yuxuan Liang](http://yuxuanliang.com), Songyu Ke, [Junbo Zhang](http://zhangjunbo.org/), Xiuwen Yi, [Yu Zheng](http://urban-computing.com/yuzheng), "[GeoMAN: Multi-level Attention Networks for Geo-sensory Time Series Prediction](https://www.ijcai.org/proceedings/2018/0476.pdf)", *IJCAI*, 2018.

If you find this code and dataset useful for your research, please cite our paper:
```
@inproceedings{ijcai2018-476,
  title     = {GeoMAN: Multi-level Attention Networks for Geo-sensory Time Series Prediction},
  author    = {Yuxuan Liang and Songyu Ke and Junbo Zhang and Xiuwen Yi and Yu Zheng},
  booktitle = {Proceedings of the Twenty-Seventh International Joint Conference on
               Artificial Intelligence, {IJCAI-18}},
  publisher = {International Joint Conferences on Artificial Intelligence Organization},
  pages     = {3428--3434},
  year      = {2018},
  month     = {7},
  doi       = {10.24963/ijcai.2018/476},
  url       = {https://doi.org/10.24963/ijcai.2018/476},
}
```

## Dataset [[Click here](http://urban-computing.com/data/Data-1.zip)]
The datasets we use for model training is detailed in Section 4.1 of our paper, which are still under processing for release. You can use the sampled data with 100 instances under "sample_data" folder. Besides, you can also test the code by using the air quality data from 2014/5/1 to 2015/4/30 in our previous research, i.e., KDD-13 and KDD-15. For more datasets, you can visit the [homepage](http://urban-computing.com/) of Urban Computing Lab in JD Group.

## Code Usage
### Preliminary
GeoMAN uses the following dependencies: 
* [TensorFlow](https://github.com/tensorflow/tensorflow#download-and-setup) >= 1.5.0
* numpy and scipy.
* CUDA 8.0 or latest version. And **cuDNN** is highly recommended.

### Code Framework
![](https://raw.githubusercontent.com/yoshall/GeoMAN/master/images/framework.png)

### Model Input
The model has the following inputs:
* local_inputs: the input of local spatial attention, shape->[batch_size, n_steps_encoder, n_input_encoder]
* global_inputs: the input of global spatial attention, shape->[batch_size, n_steps_encoder, n_sensors]
* external_inputs: the input of external factors, shape->[batch_size, n_steps_decoder, n_external_input]
* local_attn_states: shape->[batch_size, n_input_encoder, n_steps_encoder]
* global_attn_states: shape->[batch_size, n_sensors, n_input_encoder, n_steps_encoder]
* labels: ground truths, shape->[batch_size, n_steps_decoder, n_output_decoder]

### Guide
The model implement mainly lies in "GeoMAN.py" and "base_model.py" and both of them are well commented. To train or test our model, please follow the presented notebooks.

### License
GeoMAN is released under the MIT License (refer to the LICENSE file for details).