search

LibCity / Bigscity-LibCity

LibCity: An Open Library for Urban Spatial-temporal Data Mining
https://libcity.ai/
Apache License 2.0
901 stars 165 forks source link

Add model STSSL. #431

Closed Lucas-lyh closed 2 months ago

Lucas-lyh commented 3 months ago

该PR添加了STSSL模型，使用STSSL论文代码添加了model，添加了自适应loss权重的executor，添加了IN/OUT+MAE/MAPE的评估器以复现结果。

复现结果如下，标黄为复现指标

bc37e2d3c17436f195dd35a556538e9
Lucas-lyh commented 3 months ago

已修复，去除了flow预测特有的in/out分维度评测，使用通用的评测指标。