czzhangheng
/
FS-TFP
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update traffic_dataloader_v2.py

This commit is contained in:
HengZhang 2024-11-28 10:13:15 +08:00
parent b9ca3349c9
commit 50785ad3c1
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
federatedscope/trafficflow/dataloader/traffic_dataloader_v2.py
View File

@ -89,7 +89,6 @@ def load_traffic_data(config, client_cfgs):
root = config.data.root root = config.data.root
dataName = 'PEMSD' + root[-1] dataName = 'PEMSD' + root[-1]
raw_data = load_st_dataset(dataName) raw_data = load_st_dataset(dataName)
sub_graph_size = config.model.minigraph_size
l, n, f = raw_data.shape l, n, f = raw_data.shape