Skip to content

Latest commit

 

History

History
4 lines (3 loc) · 325 Bytes

README.md

File metadata and controls

4 lines (3 loc) · 325 Bytes

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.