czzhangheng
/
TrafficWheel
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

变量名更改

This commit is contained in:
czzhangheng 2025-12-06 19:48:27 +08:00
parent 865c5a3082
commit f899f50b16
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
model/AEPSA/aepsav3.py
View File

@ -190,7 +190,7 @@ class AEPSA(nn.Module):
# 图编码
H_t = self.graph_encoder(x_enc) # [B,N,1,T] -> [B, N, hidden_dim*(K+1)]
X_t_1 = self.graph_projection(H_t) # [B,N,d_model]
enc_out = torch.cat([H_t, X_t_1], dim = -1) # [B, N, d_model + hidden_dim*(K+1)]
X_enc = torch.cat([H_t, X_t_1], dim = -1) # [B, N, d_model + hidden_dim*(K+1)]
# 词嵌入处理
self.mapping_layer(self.word_embeddings.permute(1, 0)).permute(1, 0)
@ -198,9 +198,9 @@ class AEPSA(nn.Module):
source_embeddings = self.word_embeddings[masks==1] # [selected_words,d_llm]
# 重编程与预测
enc_out = self.reprogramming_layer(enc_out, source_embeddings, source_embeddings)
enc_out = self.gpts(inputs_embeds=enc_out).last_hidden_state # [B,N,d_llm]
dec_out = self.out_mlp(enc_out) # [B,N,pred_len]
X_enc = self.reprogramming_layer(X_enc, source_embeddings, source_embeddings)
X_enc = self.gpts(inputs_embeds=X_enc).last_hidden_state # [B,N,d_llm]
dec_out = self.out_mlp(X_enc) # [B,N,pred_len]
# 维度调整
outputs = dec_out.unsqueeze(dim=-1) # [B,N,pred_len,1]