Skip to content

Commit

Permalink
fix: data embedding input dim should be n_features;
Browse files Browse the repository at this point in the history
  • Loading branch information
WenjieDu committed Apr 2, 2024
1 parent 8708853 commit 43a4ecc
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pypots/imputation/informer/modules/core.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def __init__(
self.seq_len = n_steps
self.n_layers = n_layers
self.enc_embedding = DataEmbedding(
n_features * 2,
n_features,
d_model,
dropout=dropout,
)
Expand Down

0 comments on commit 43a4ecc

Please sign in to comment.