ZhuangDingyi / STZINB

Source code of implementing spatial-temporal zero-inflated negative binomial network for trip demand prediction
MIT License
24 stars 7 forks source link

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.