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

[fix] extend SSTBAN input/output dim from 1 to data dim #401

Open Blore-lzn opened 6 months ago

Blore-lzn commented 6 months ago

调整SSTBAN数据集输入/输出维度从固定值1为数据本身维度/output_dim

已在NYCBike1/NYCTaxi/BJTaxi上进行测试

问题：目前SSTBAN的配置文件中有in_channels和out_channels两个参数，实际对应了模型的输入维度和输出维度，建议将这两个与feature_dim/output_dim参数合并