在研究STGCN中遇到的问题

最近再看STGCN的代码,其中对于gso_type这个参数不太明白,我猜测其为领接矩阵的格式,但我不明白这些代表什么意思,希望有解答
parser.add_argument('--gso_type', type=str, default='sym_norm_lap', choices=['sym_norm_lap', 'rw_norm_lap', 'sym_renorm_adj', 'rw_renorm_adj'])

参考GPT和自己的思路:

您的猜测是正确的,gso_type是表示领接矩阵的格式。具体来说,这个参数对应的四个选项含义如下:

  1. sym_norm_lap:表示对称归一化的拉普拉斯矩阵
  2. rw_norm_lap:表示随机游走归一化的拉普拉斯矩阵
  3. sym_renorm_adj:表示对称重归一化的邻接矩阵
  4. rw_renorm_adj:表示随机游走重归一化的邻接矩阵

不同的领接矩阵格式对应了不同的图卷积网络模型。选择哪种格式应当基于具体的实验和数据情况而定。

gso_type表示领接矩阵的类型