上一篇文章GC-LSTM原理部分中具体介绍了GC-LSTM的原理,这里基于PyG Temporal中的GCLSTM模块对实现一个基于GC-LSTM的预测模型。注:这里可以是流量预测等各种节点特征。
补充说明:GCLSTM模型中使用的是切比雪夫滤波器(多项式滤波器),我之前忽略了这一点,下面的模型中,大家可以根据需要将参数K进行调试。
一、数据预处理
这里图拓扑的结构和邻接矩阵如图所示,:
节点特征矩阵,这里只展示了其中一个节点的特征向量随时间的变化,只选取了一个指标即:
接下来我们需要对数据进行预处理,具体过程如代码所示:
def clean_train_test_data(traindata, testdata):
sc = M