LibCity (阡陌) is a unified, comprehensive and extensible library for traffic prediction, which provides researchers with a credible experimental tool and a convenient development framework. Our library is implemented based on PyTorch and includes all the necessary steps or components related on traffic prediction into a systematic pipeline.
LibCity currently supports following tasks:
2. The LibCity Library
The overall framework of LibCity is presented as below, consisting of five major modules. It provides various datasets, mechanisms, models and utilities to support data preprocessing, model instantiation and performance evaluation for the four kinds of tasks.
Overview of the LibCity library
The main features of LibCity can be summarized in three aspects:
The reproduced traffic prediction models are categorized in the below table.
The implemented models in LibCity
3. Contact Us
LibCity is mainly developed and maintained by Beihang Interest Group on SmartCity (BIGSCITY). Welcome to visit and use our GitHub repository and website for more details. Your suggestions and contributions are very important to us! If you have any question about the library, please raise an issue in our GitHub repository.