(self, configs)
| 21 | |
| 22 | class ScaleGraphBlock(nn.Module): |
| 23 | def __init__(self, configs): |
| 24 | super(ScaleGraphBlock, self).__init__() |
| 25 | self.seq_len = configs.seq_len |
| 26 | self.pred_len = configs.pred_len |
| 27 | self.k = configs.top_k |
| 28 | |
| 29 | self.att0 = Attention_Block(configs.d_model, configs.d_ff, |
| 30 | n_heads=configs.n_heads, dropout=configs.dropout, activation="gelu") |
| 31 | self.norm = nn.LayerNorm(configs.d_model) |
| 32 | self.gelu = nn.GELU() |
| 33 | self.gconv = nn.ModuleList() |
| 34 | for i in range(self.k): |
| 35 | self.gconv.append( |
| 36 | GraphBlock(configs.c_out , configs.d_model , configs.conv_channel, configs.skip_channel, |
| 37 | configs.gcn_depth , configs.dropout, configs.propalpha ,configs.seq_len, |
| 38 | configs.node_dim)) |
| 39 | |
| 40 | |
| 41 | def forward(self, x): |
nothing calls this directly
no test coverage detected