ZhuangDingyi / STZINB

Source code of implementing spatial-temporal zero-inflated negative binomial network for trip demand prediction

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

STZINB

Source code of implementing spatial-temporal zero-inflated negative binomial network for trip demand prediction

Run python main_zero_inflated.py to train the model and generate model prediction and evaluation using python test_zero_inflated.py. Remember to replace the corresponding data files and output files.

About

Source code of implementing spatial-temporal zero-inflated negative binomial network for trip demand prediction

License:MIT License


Languages

Language:Python 100.0%