关于PeMSD4数据集跑模型遇到的相关问题 about bigscity-libcity HOT 6 CLOSED

你好，关于这个问题主要考虑到PEMSD4数据集的特殊性，1是它本身包含交通速度流量和占有率3维数据，而一般的模型则适用于跑其中一维数据，不适合全部数据一起跑。这个可以从PEMSD4/config.json里边的data_col参数进行配置（我们建议您设置data_col: traffic_speed这样再进行实验），选择你想要跑哪一维的数据，2这个数据集的图结构中存在入度为0节点。

from bigscity-libcity.

另外，针对您说的4个模型，
对于STMGAT模型，他就是由于图结构的问题报错的，您可以在代码libcity/model/traffic_speed_prediction/STMGAT.py第111行的g.add_edges(src, dst)后边加上g = dgl.add_self_loop(g)自环，来解决入度为0的问题。

对于ATDM模型，在我们的测试中，他在PEMSD4数据集上运行没有问题，需要您提供具体的配置。

对于HGCN模型和DKFN模型，建议先使用METR_LA数据集进行实验，这两个模型的问题我们会在近期的更新中解决。

from bigscity-libcity.

你好，关于这个问题主要考虑到PEMSD4数据集的特殊性，1是它本身包含交通速度流量和占有率3维数据，而一般的模型则适用于跑其中一维数据，不适合全部数据一起跑。这个可以从PEMSD4/config.json里边的data_col参数进行配置（我们建议您设置data_col: traffic_speed这样再进行实验），选择你想要跑哪一维的数据，2这个数据集的图结构中存在入度为0节点。

感谢您的回复，但是设置完data_col: traffic_speed，进行实验时，提示ValueError("batch['y_true'].shape is not equal to batch['y_pred'].shape")，请问还需要在哪里进行更改嘛

from bigscity-libcity.

你好，关于这个问题主要考虑到PEMSD4数据集的特殊性，1是它本身包含交通速度流量和占有率3维数据，而一般的模型则适用于跑其中一维数据，不适合全部数据一起跑。这个可以从PEMSD4/config.json里边的data_col参数进行配置（我们建议您设置data_col: traffic_speed这样再进行实验），选择你想要跑哪一维的数据，2这个数据集的图结构中存在入度为0节点。

感谢您的回复，但是设置完data_col: traffic_speed，进行实验时，提示ValueError("batch['y_true'].shape is not equal to batch['y_pred'].shape")，请问还需要在哪里进行更改嘛

哪一个模型呢？对了，记得同时设置output_dim=1，因为如果输入只有速度，那么预测的输出维度也应该只有1维。
另外，修改参数后，记得在运行之前删除掉libcity/cache/dataset_cache/目录下边缓存的数据文件，否则很可能会继续加载之前的数据，导致报错。

from bigscity-libcity.

你好，关于这个问题主要考虑到PEMSD4数据集的特殊性，1是它本身包含交通速度流量和占有率3维数据，而一般的模型则适用于跑其中一维数据，不适合全部数据一起跑。这个可以从PEMSD4/config.json里边的data_col参数进行配置（我们建议您设置data_col: traffic_speed这样再进行实验），选择你想要跑哪一维的数据，2这个数据集的图结构中存在入度为0节点。

感谢您的回复，但是设置完data_col: traffic_speed，进行实验时，提示ValueError("batch['y_true'].shape is not equal to batch['y_pred'].shape")，请问还需要在哪里进行更改嘛

哪一个模型呢？对了，记得同时设置output_dim=1，因为如果输入只有速度，那么预测的输出维度也应该只有1维。 另外，修改参数后，记得在运行之前删除掉libcity/cache/dataset_cache/目录下边缓存的数据文件，否则很可能会继续加载之前的数据，导致报错。

好的 我改一下试试

from bigscity-libcity.

DKFN,STMGAT,HGCN的问题均已解决

from bigscity-libcity.