The code of MultiSPANS: A Multi-range Spatial-Temporal Transformer Network for Traffic Forecast via Structural Entropy Optimization, which is accepted by WSDM2024.
The core code is located at ./TrafficForecast/MultiSPANS/libcity/model/traffic_flow_prediction The project can be fully adapted to Libcity benckmark.
The implementation of MultiSPANS is under Python 3.10, with the following key packages installed:
pytorch == 1.12.0
torch-geometric == 2.3.0