bigscity/ST-nFBST

[IJCAI2024] A PyTorch implementation of Full Bayesian Signficance Testing for Neural Networks in Traffic Forecasting

This is a PyTorch implementation of Full Bayesian Significance Testing for Neural Networks in Traffic Forecasting, based on LibCity/Bigscity-LibCity: LibCity: An Open Library for Urban Spatial-temporal Data Mining (github.com).

The core code of ST-nFBST is in libcity/model/traffic_speed_prediction/BDCRNNVariableDecoder.py.

References:

Install and quick start — Bigscity-LibCity documentation (bigscity-libcity-docs.readthedocs.io)